Skip to content

Commit

Permalink
Update video file name processing to be platform agnostic.
Browse files Browse the repository at this point in the history
  • Loading branch information
bk committed Oct 12, 2018
1 parent d29e3af commit 9264b66
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion video_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
from yolo.util import write_results, dynamic_write_results
from SPPE.src.main_fast_inference import *

import ntpath
import os
import sys
from tqdm import tqdm
Expand Down Expand Up @@ -62,7 +63,7 @@
}

# Data writer
save_path = os.path.join(args.outputpath, 'AlphaPose_'+videofile.split('/')[-1].split('.')[0]+'.avi')
save_path = os.path.join(args.outputpath, 'AlphaPose_'+ntpath.basename(videofile).split('.')[0]+'.avi')
writer = DataWriter(args.save_video, save_path, cv2.VideoWriter_fourcc(*'XVID'), fps, frameSize).start()

im_names_desc = tqdm(range(data_loader.length()))
Expand Down

0 comments on commit 9264b66

Please sign in to comment.