Agregar una función a DeOldify (Colorización por IA)

Quiero hacer que éste programa de código abierto (convierte videos blanco y negro a color) pueda trabajar indicándole colores puntuales, para que no los tenga que "inventar".

Ésto podría ser aportándole uno o más fotogramas coloreados previamente de manera manual, o simplemente adjuntando una foto o hasta una paleta con los colores deseados. Permitiendo así que el algoritmo aprenda sobre esas imágenes y recomiende de acuerdo a lo que el usuario busca.

Complementando de ésta forma el programa se podrían obtener resultados mucho más sólidos en cuanto a colorización, ya que su principal problema es que no termina de "decidirse" por algunos colores que le resultan ambiguos, entonces los videos suelen salir con fluctuaciones que no son convincentes.

Busco programador avanzado, preferentemente argentino, que maneje aprendizaje automático, machine learning, ciencia de datos, visión artificial, librerías de tensores y python, entre otras cosas.

Forma de pago a convenir, en pesos argentinos o dólares.