其实这种网络很容易实现, 你只需要灰色的训练图片, 和彩色的groundtruth即可.
至于网络如何构建, 可以参考分割网络的实现. 也就说, 网络直接输出一张(3, 256, 256)的张量. 我之前在博客里面有实现过一个训练单张图片的, 其实那是一个最简单的版本, 试想一下, 一张图片的输入输出与一个batch并没有本质的区别.
过段时间我会更新一个更加前沿的encoder-decoder架构的灰色变彩色AI并开源到MANAAI, 欢迎持续关注
与 神励AI工坊 的连接断开,我们正在尝试重连,请耐心等待