This commit is contained in:
Kaushik Narayan R 2023-10-15 17:52:51 -07:00
parent 3f269f10e8
commit 0a12f0fd74
2 changed files with 2 additions and 2 deletions

View File

@ -163,7 +163,7 @@
" image_fd = np.array(get_all_fd(image_id)[selected_feature_model]).flatten()\n", " image_fd = np.array(get_all_fd(image_id)[selected_feature_model]).flatten()\n",
" else:\n", " else:\n",
" image_fd = np.array(\n", " image_fd = np.array(\n",
" fd_collection.find({\"image_id\": image_id})[selected_feature_model]\n", " fd_collection.find_one({\"image_id\": image_id})[selected_feature_model]\n",
" ).flatten()\n", " ).flatten()\n",
"\n", "\n",
" match dim_reduction:\n", " match dim_reduction:\n",

View File

@ -163,7 +163,7 @@
" image_fd = np.array(get_all_fd(image_id)[selected_feature_model]).flatten()\n", " image_fd = np.array(get_all_fd(image_id)[selected_feature_model]).flatten()\n",
" else:\n", " else:\n",
" image_fd = np.array(\n", " image_fd = np.array(\n",
" fd_collection.find({\"image_id\": image_id})[selected_feature_model]\n", " fd_collection.find_one({\"image_id\": image_id})[selected_feature_model]\n",
" ).flatten()\n", " ).flatten()\n",
"\n", "\n",
" match dim_reduction:\n", " match dim_reduction:\n",