diff --git a/detect.py b/detect.py index 24cbfa02..5428328d 100755 --- a/detect.py +++ b/detect.py @@ -106,7 +106,7 @@ def detect( if save_images: # Save generated image with detections - cv2.imwrite(results_img_path.replace('.bmp', '.jpg').replace('.tif', '.jpg'), img) + cv2.imwrite(results_img_path, img) print(' Done. (%.3fs)' % (time.time() - t))