Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -99,10 +99,9 @@ def perform_object_detection(image, model_name, threshold):
|
|
| 99 |
|
| 100 |
|
| 101 |
# Perform object detection and draw bounding boxes
|
| 102 |
-
|
| 103 |
processed_image, detected_objects = detect_and_draw_objects(image, model_name, threshold)
|
| 104 |
-
|
| 105 |
-
st.write(detected_objects[:20])
|
| 106 |
return processed_image, detected_objects
|
| 107 |
|
| 108 |
|
|
@@ -113,25 +112,20 @@ if detect_button:
|
|
| 113 |
# Open the uploaded image
|
| 114 |
try:
|
| 115 |
image = Image.open(image)
|
| 116 |
-
|
| 117 |
# Display the original image
|
| 118 |
st.image(image, use_column_width=True, caption="Original Image")
|
| 119 |
-
|
| 120 |
# Perform object detection
|
| 121 |
processed_image, detected_objects = perform_object_detection(image, detect_model, threshold)
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
st.write(processed_image)
|
| 125 |
-
st.write(type(processed_image))
|
| 126 |
-
#if processed_image:
|
| 127 |
# Display the image with detected objects
|
| 128 |
st.image(processed_image, use_column_width=True, caption="Image with Detected Objects")
|
| 129 |
-
|
| 130 |
# Display the detected objects as text
|
| 131 |
st.write(detected_objects)
|
| 132 |
-
|
| 133 |
-
#else:
|
| 134 |
-
# st.error("Failed to process image for object detection.")
|
| 135 |
|
| 136 |
except Exception as e:
|
| 137 |
st.error(f"Error loading image: {e}")
|
|
|
|
| 99 |
|
| 100 |
|
| 101 |
# Perform object detection and draw bounding boxes
|
| 102 |
+
|
| 103 |
processed_image, detected_objects = detect_and_draw_objects(image, model_name, threshold)
|
| 104 |
+
|
|
|
|
| 105 |
return processed_image, detected_objects
|
| 106 |
|
| 107 |
|
|
|
|
| 112 |
# Open the uploaded image
|
| 113 |
try:
|
| 114 |
image = Image.open(image)
|
| 115 |
+
|
| 116 |
# Display the original image
|
| 117 |
st.image(image, use_column_width=True, caption="Original Image")
|
| 118 |
+
|
| 119 |
# Perform object detection
|
| 120 |
processed_image, detected_objects = perform_object_detection(image, detect_model, threshold)
|
| 121 |
+
|
| 122 |
+
|
|
|
|
|
|
|
|
|
|
| 123 |
# Display the image with detected objects
|
| 124 |
st.image(processed_image, use_column_width=True, caption="Image with Detected Objects")
|
| 125 |
+
|
| 126 |
# Display the detected objects as text
|
| 127 |
st.write(detected_objects)
|
| 128 |
+
|
|
|
|
|
|
|
| 129 |
|
| 130 |
except Exception as e:
|
| 131 |
st.error(f"Error loading image: {e}")
|