未验证 提交 ac4b41ea 编写于 作者: L LielinJiang 提交者: GitHub

fix model bug (#64)

上级 1b84c4fb
......@@ -47,7 +47,7 @@ class SingleDataset(BaseDataset):
A_paths(str) - - the path of the image
"""
A_path = self.A_paths[index]
A_img = cv2.imread(A_path)
A_img = cv2.cvtColor(cv2.imread(A_path), cv2.COLOR_BGR2RGB)
A = self.transform(A_img)
return {'A': A, 'A_paths': A_path}
......
......@@ -135,8 +135,7 @@ class NLayerDiscriminator(nn.Layer):
1,
kernel_size=kw,
stride=1,
padding=padw,
bias_attr=False)
padding=padw)
] # output 1 channel prediction map
self.model = nn.Sequential(*sequence)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册