unet-pytorch
unet-pytorch copied to clipboard
博主您好,我做二分类使用您的train_Medical训练,效果很差
我想问一下,你的数据medical数据预处理modify_png[png <= 127.5] = 1
,用自己的数据集这里需要修改一下吗:
`pg, png = self.get_random_data(jpg, png, self.input_shape, random = self.train)
jpg = np.transpose(preprocess_input(np.array(jpg, np.float64)), [2,0,1])
png = np.array(png)
#-------------------------------------------------------#
# 这里的标签处理方式和普通voc的处理方式不同
# 将小于127.5的像素点设置为目标像素点。
#-------------------------------------------------------#
modify_png = np.zeros_like(png)
modify_png[png <= 127.5] = 1
seg_labels = modify_png
seg_labels = np.eye(self.num_classes + 1)[seg_labels.reshape([-1])]
seg_labels = seg_labels.reshape((int(self.input_shape[0]), int(self.input_shape[1]), self.num_classes + 1))
`
这和你的数据集有关呀。
我的意思,用自己的数据集,这里需要修改吗modify_png[png <= 127.5] = 1
这和你的数据集格式有关………………
可以解释一下,什么样的格式数据集需要modify_png[png <= 127.5] = 1