DMT-Net
DMT-Net copied to clipboard
liuye123321
Image dehazing
您好，读了您的论文，然后在这找到了您的代码，现在存在一个问题就是我无法使用您的代码对单张图像做测试， 然后也没有找到您的模型地址，请问您有公布您的模型吗？ 能否发我一个做模型效果测试。 非常感谢， 祝好 tongchangD
RuntimeError: Given input size: (8x1x1). Calculated output size: (8x0x0). Output size is too small
在运行main.py的时候报了这个错，用的是Haze4K数据集，看了下代码发现应该是common.py文件中的第277行class Down有问题，因为atmosphere map为size为(8,1,1,1)，而代码里写了Maxpool2d(2)所以报错，我现在自己这里改了下对于atmosphere map设置了个检测，如果是atmosphere map就Maxpool2d(1)，想问下作者你跑通的代码是怎么写的，我这么写会影响实验么
` elif is_jpg_file(input_path):` `AB = cv2.imread(input_path).astype("float") / 255.0` `haze_image, GT = get_patch_da(AB, self.size)` `trans = np.zeros((self.size, self.size, 3))` 这段代码应该是DAD的6000张数据集的代码，DAD应该是有transmission map的，然后没有atmospheric map。但程序里好像是对transmission map和atmospheric map设置为0，但这样与论文里关于3.2 Supervised Loss on Labeled data 有误差，而且这样计算transmission loss...
