Commit e20fad34 authored by vaisakh.nair's avatar vaisakh.nair 🎯

new changes

parent f726bc6e
from yolov7processor import YoloV7Processor 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 = 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() detect.detect()
...@@ -231,11 +231,11 @@ class YoloV7Processor(): ...@@ -231,11 +231,11 @@ class YoloV7Processor():
# vid_writer = cv2.VideoWriter(save_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h)) # vid_writer = cv2.VideoWriter(save_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h))
# vid_writer.write(im0) # vid_writer.write(im0)
# #
# if save_txt or save_img: 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 '' 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"Results saved to {save_dir}{s}")
#
# print(f'Done. ({time.time() - t0:.3f}s)') print(f'Done. ({time.time() - t0:.3f}s)')
# #
# #
# if __name__ == '__main__': # if __name__ == '__main__':
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment