Update app.py
Browse files
app.py
CHANGED
@@ -152,8 +152,8 @@ def detect(opt, save_img=False):
|
|
152 |
# Save results (image with detections)
|
153 |
if save_img:
|
154 |
if dataset.mode == 'image':
|
155 |
-
Image.fromarray(im0).show()
|
156 |
-
im0.save(save_path)
|
157 |
# cv2.imwrite(save_path, im0)
|
158 |
print(f" The image with the result is saved in: {save_path}")
|
159 |
# else: # 'video' or 'stream'
|
|
|
152 |
# Save results (image with detections)
|
153 |
if save_img:
|
154 |
if dataset.mode == 'image':
|
155 |
+
# Image.fromarray(im0).show()
|
156 |
+
Image.fromarray(im0).save(save_path)
|
157 |
# cv2.imwrite(save_path, im0)
|
158 |
print(f" The image with the result is saved in: {save_path}")
|
159 |
# else: # 'video' or 'stream'
|