https://github.com/yinboc/liif/blob/main/models/liif.py
满大街hardcode参数,大堆看不懂的轴,大堆莫名其妙的变量缩写。。。
看来玩这个图像坐标相关(并xy与外循环),又要新库去操作轴了
imnet_spec:
name: mlp
out_dim: 3
hidden_list: [256, 256, 256, 256]
self.encoder = models.make(encoder_spec)
self.imnet = models.make(imnet_spec, args={
'in_dim': imnet_in_dim})