Update utils/video_process.py

utils/video_process.py

@@ -31,7 +31,7 @@ def process_tse(args, model, device, data_reader, output_wave_dir):
 	with torch.no_grad():
 		for videoPath in data_reader:  # Loop over all video samples
 			savFolder = videoPath.split('/')[-1]
-			video_args.savePath = f'{output_wave_dir}/{savFolder[:-4]}/'
+			video_args.savePath = f'{output_wave_dir}/{savFolder.split(".")[0]}/'
 			video_args.videoPath = videoPath
 			main(video_args, args)