Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-02-24 12:29:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #71 from chappjo/main

Browse Source 
Add support for edge cases like outputting 3gp file container
This commit is contained in:
hzwer
2020-12-14 18:10:46 +08:00
committed by GitHub
parent f284bd409d 17672ca0a0
commit c6e66ea757
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
inference_video.py
View File

@@ -36,10 +36,16 @@ def transferAudio(sourceVideo, targetVideo):
tempAudioFileName = "./temp/audio.mp3"
os.system("ffmpeg -y -i " + sourceVideo + " -c:a mp3 -vn " + tempAudioFileName)
os.system("ffmpeg -y -i " + "noAudio_"+targetVideo + " -i " + tempAudioFileName + " -c copy " + targetVideo)
print("Lossless audio transfer failed. Audio was transcoded to mp3 instead.")
if (os.path.getsize(targetVideo) == 0): # if mp3 not supported by selected format
os.rename("noAudio_"+targetVideo, targetVideo)
print("Audio transfer failed. Interpolated video will have no audio")
else:
print("Lossless audio transfer failed. Audio was transcoded to mp3 instead.")
# remove audio-less video
os.remove("noAudio_"+targetVideo)
# remove audio-less video
os.remove("noAudio_"+targetVideo)
else:
os.remove("noAudio_"+targetVideo)
# remove temp directory
shutil.rmtree("temp")
@@ -210,3 +216,4 @@ if args.png == False and fpsNotAssigned == True and not args.skip and not args.v
transferAudio(args.video, outputVideoFileName)
except:
print("Audio transfer failed. Interpolated video will have no audio")
os.rename("noAudio_"+outputVideoFileName, outputVideoFileName)