first

DetectionVideo_Video.py

@@ -27,7 +27,7 @@ from vediotoima import vedio_img
 from imatovideo import images_to_video,images_to_video_big
 from moviepy.editor import *
 import os
-os.environ['CUDA_VISIBLE_DEVICES'] = '2'
+os.environ['CUDA_VISIBLE_DEVICES'] = '3'
 
 def parse_args():
 
@@ -39,10 +39,11 @@ def parse_args():
         help="path to config file",
         type=str,
     )
-    parser.add_argument("--ckpt", type=str, default="../data_MPKFNet/weigth/detection/detection.pth",
-                        help="Trained weights.")
+    #.../data_MPKFNet/weigth/detection/detection.pth
+   # parser.add_argument("--ckpt", type=str, default="../data_MPKFNet/weigth/detection/weigth_all/detectionboyanghu.pth",
+                        #help="Trained weights.")
     parser.add_argument("--score_threshold", type=float, default=0.3)
-    parser.add_argument("--video_name", default='000120210501080936.mp4', type=str, help='Video name.')
+    parser.add_argument("video_name", default='YNhiv00016.mp4', type=str, help='Video name.')
     parser.add_argument("--images_dir", default='../data_MPKFNet/result_raw/yang/', type=str,
                         help='Specify a image dir to do prediction.')
     parser.add_argument("--output_dir", default='../data_MPKFNet/result/yangd/', type=str,
@@ -111,7 +112,7 @@ if __name__ == '__main__':
     args = parse_args()
     video_name_path = args.video_name
     video_name_path = video_name_path.replace(".MP4", "").replace(".mp4", "").replace(".MOV", "").replace(".mov", "")
-    RawVideoFilename = '../data_MPKFNet/rawdata/1/'+args.video_name
+    RawVideoFilename = '../data_MPKFNet/rawdata/'+args.video_name
     RawVideoExtrationSavedpath = '../data_MPKFNet/result_raw/' + video_name_path
     args.images_dir =  '../data_MPKFNet/result_raw/' + video_name_path
     args.output_dir = "../data_MPKFNet/result/" + video_name_path + "Detection"