add_model_resnet18 #31

Merged
brian merged 47 commits from add_model_resnet18 into main 2024-04-03 21:04:23 +02:00
Showing only changes of commit 080c7e7a5c - Show all commits
+4 -1
View File
@@ -37,6 +37,7 @@ def list_names(
query = {}
if has_annotation:
query['annotation'] = {'$ne': None}
# execute query
res_list = collection.find(
filter=query,
projection={
@@ -44,7 +45,9 @@ def list_names(
'name': True,
},
)
return list(res_list)
# extract information
name_list = [elem['name'] for elem in res_list]
return name_list
def total_documents(