Spaces:
Sleeping
Sleeping
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 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|