Colorization with averaging ab channels Link to train repository /~https://github.com/IlyaKodua/colorization_with_mask Link to weigths of colorizator https://drive.google.com/file/d/1m6TTc0K8v-d4hFJBjyPHHO3HdBU7Qs4S/view train scheme EXAMPLE real image averaged mask output image