未验证 提交 395b130f 编写于 作者: L Lv Mengsi 提交者: GitHub

fix spade (#3964)

上级 fe24e719
......@@ -308,7 +308,7 @@ class triplex_reader_creator(reader_creator):
input_label = np.zeros(
(args.label_nc, index.shape[1], index.shape[2]))
np.put_along_axis(input_label, index, 1.0, 0)
img1 = input_label
img1 = input_label.astype('float32')
img2 = (np.array(img2).astype('float32') / 255.0 - 0.5) / 0.5
img2 = img2.transpose([2, 0, 1])
if not args.no_instance:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册