Spaces:
Running
Running
Update my_model/object_detection.py
Browse files
my_model/object_detection.py
CHANGED
@@ -245,15 +245,7 @@ def detect_and_draw_objects(image_path, model_type='yolov5', threshold=0.2, show
|
|
245 |
return image_with_boxes, detected_objects_string
|
246 |
|
247 |
|
248 |
-
|
249 |
if __name__ == "__main__":
|
250 |
pass
|
251 |
|
252 |
-
# 'Sample_Images' is the folder conatining sample images for demo.
|
253 |
-
image_path = get_path('horse.jpg', 'Sample_Images')
|
254 |
-
processed_image, objects_string = detect_and_draw_objects(image_path,
|
255 |
-
model_type='detic',
|
256 |
-
threshold=0.2,
|
257 |
-
show_confidence=False)
|
258 |
-
show_image(processed_image)
|
259 |
-
print("Detected Objects:", objects_string)
|
|
|
245 |
return image_with_boxes, detected_objects_string
|
246 |
|
247 |
|
248 |
+
|
249 |
if __name__ == "__main__":
|
250 |
pass
|
251 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|