new changes

app.py

 from yolov7processor import YoloV7Processor
 
-detect = YoloV7Processor(source =r'D:\JK_Data\videos\new_Data\camera_46\2023-03-26_05-12-30_5', weights =r'D:\JK_Data\yolov7-tiny-v5\weights\jk_v5.pt', imgsz = 416, trace = True, target_dir=r'D:\JK_Data\videos\new_Data\camera_46\2023-03-26_05-12-30_5')
-detect.detect()
+detect = YoloV7Processor(source =r'D:\JK_Data\overlap_retrain_dataset\loader -7\loader_7_frames', weights =r'D:\JK_Data\yolov7_models\yolov7-tiny-v5\weights\jk_v5.pt', imgsz = 416, trace = True, target_dir=r'D:\JK_Data\overlap_retrain_dataset\loader -7\loader_7_frames')
+detect.detect() 
\ No newline at end of file

yolov7processor.py

@@ -231,11 +231,11 @@ class YoloV7Processor():
         #                     vid_writer = cv2.VideoWriter(save_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h))
         #                 vid_writer.write(im0)
         #
-        # if save_txt or save_img:
-        #     s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
-        #     #print(f"Results saved to {save_dir}{s}")
-        #
-        # print(f'Done. ({time.time() - t0:.3f}s)')
+        if save_txt or save_img:
+            s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
+            #print(f"Results saved to {save_dir}{s}")
+        
+        print(f'Done. ({time.time() - t0:.3f}s)')
         #
     #
     # if __name__ == '__main__':