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

new changes

parent f726bc6e
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()
\ No newline at end of file
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()
......@@ -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__':
......
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