Fix predictor bug on windows (PaddlePaddle#210)

* update install docs * fix windows bug * fix download when in windows * fix video2frames bug on windows platform
AnnaTrainingG · Mar 4, 2021 · 4183009 · 4183009
1 parent 62ce17f
commit 4183009
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ppgan/apps/dain_predictor.py b/ppgan/apps/dain_predictor.py
@@ -78,7 +78,7 @@ def run(self, video_path):
 
         out_path = video2frames(video_path, frame_path_input)
 
-        vidname = video_path.split(os.sep)[-1].split('.')[0]
+        vidname = os.path.basename(video_path).split('.')[0]
 
         frames = sorted(glob.glob(os.path.join(out_path, '*.png')))
 

diff --git a/ppgan/utils/video.py b/ppgan/utils/video.py
@@ -24,7 +24,7 @@ def _dict2str(kargs):
         return cmd_str
 
     ffmpeg = ['ffmpeg ', ' -y -loglevel ', ' error ']
-    vid_name = video_path.split(os.sep)[-1].split('.')[0]
+    vid_name = os.path.basename(video_path).split('.')[0]
     out_full_path = os.path.join(outpath, vid_name)
 
     if not os.path.exists(out_full_path):