jantic / DeOldify

A Deep Learning based project for colorizing and restoring old images (and video!)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

video building only works for mp4s

meksor opened this issue · comments

Currently, as far as i can see, rebuilding the video (from the colorized frames) only works if the original input video is a .mp4. (I think that is the reason for #471 as i also have a .mkv based workflow) I have a fix for this on my fork which i would be willing to split up into a PR.
However, I am not very familiar with this project so I wanted to ask if this is a desired feature and if there is any reason to blacklist or whitelist specific video formats.

Regards

PS: the culprit:

source_path.name.replace('.mp4', '_no_audio.mp4')

Yeah that's a deliberate decision. As the aim is to just make research freely available to the public, I made a few simplifying decisions to keep maintenance and bugs to a minimum. This is one of them.

I understand. It might be good to add a check and error message otherwise the decision does not seem too deliberate.
Anyway I will keep using my fork then, I completely understand not wanting to put effort into this.