first

DensityVideo_Video.py

@@ -129,11 +129,11 @@ def inf(DensityEstimationResultPath,devices):
         with open (json_outputpath,'w',encoding='utf-8') as f:
             f.write(str(json_data))
 
-        # plt.figure(num_display)
-        # plt.title("predict:{}".format(int(num_display)))
-        # plt.imshow(targets_pred1[0, 0, :, :])
-        # plt.savefig(DensityEstimationResultPath + "/" + str(img_pathtemp))
-        # plt.close()
+        plt.figure(num_display)
+        plt.title("predict:{}".format(int(num_display)))
+        plt.imshow(targets_pred1[0, 0, :, :])
+        plt.savefig(DensityEstimationResultPath + "/" + str(img_pathtemp))
+        plt.close()
 
         i += 1
 
@@ -170,7 +170,7 @@ if __name__ == '__main__':
                         help='Video name.')
     parser.add_argument("skip_getimg", default=False, type=str,
                         help='skip_getimg.')
-    parser.add_argument("GPU", default='0', type=str,
+    parser.add_argument("GPU", default='6', type=str,
                         help='GPU.')
 
     parser.add_argument(
@@ -195,7 +195,7 @@ if __name__ == '__main__':
     CombineVideoResultPath =      '../data_MPKFNet/result_vedio/'+video_name_path+"desity.avi"
     print(args.skip_getimg)
     print(args.GPU)
-    if not args.skip_getimg:
+    if  args.skip_getimg == "False":
 
         vedio_img(RawVideoFilename, RawVideoExtrationSavedpath, 1)
     get_name = []