diff --git a/object_detection/object_detection.ipynb b/object_detection/object_detection.ipynb index 0b47ae9..7f76056 100644 --- a/object_detection/object_detection.ipynb +++ b/object_detection/object_detection.ipynb @@ -603,11 +603,11 @@ "source": [ "for c in class_names.keys():\n", " class_num = int(c)\n", - " sorted_indices = np.argsort(areas_3)[::-1]\n", + " sorted_indices = np.argsort(max_deviation_values)[::-1]\n", " count = 0\n", "\n", " for image_to_visualize in sorted_indices:\n", - " if areas_3[i] == 0 or class_nm[i] != class_num:\n", + " if max_deviation_values[i] == 0 or max_deviation_classes[i] != class_num:\n", " continue\n", " image_path = IMAGE_PATH + labels[image_to_visualize]['seg_map']\n", " print(image_path, '| idx', image_to_visualize , '| label quality score:', scores[image_to_visualize], '| is issue:', image_to_visualize in issue_idx)\n",