Update test_video_ava.py

test_video_ava.py

@@ -26,7 +26,7 @@ torch.backends.cudnn.enabled = False
 #dist.init_process_group(backend='nccl')
 model_alert = SimpleANN()
 
-model_alert.load_state_dict(torch.load(r'D:\yowov2V7\YOWOv2\model_weights.pth'))
+model_alert.load_state_dict(torch.load(r'.\model_weights.pth'))
 
 model_alert.eval()
 
@@ -38,7 +38,7 @@ def parse_args():
                         help='the size of input frame')
     parser.add_argument('--show', action='store_true', default=False,
                         help='show the visulization results.')
-    parser.add_argument('--cuda', action='store_true', default=False, 
+    parser.add_argument('--cuda', action='store_true', default=True, 
                         help='use cuda.')
     parser.add_argument('--save_folder', default='det_results/', type=str,
                         help='Dir to save results')
@@ -52,7 +52,7 @@ def parse_args():
     # model
     parser.add_argument('-v', '--version', default='yowo_v2_large', type=str,
                         help='build YOWOv2')
-    parser.add_argument('--weight', default='weight/',
+    parser.add_argument('--weight', default='./backup_dir/ava_v2.2/fps32_k16_bs16_yolo_large_newdata_p2/epoch4/yowo_v2_large_epoch_4.pth',
                         type=str, help='Trained state_dict file path to open')
     parser.add_argument('--topk', default=40, type=int,
                         help='NMS threshold')