unet-pytorch icon indicating copy to clipboard operation
unet-pytorch copied to clipboard

博主您好,我做二分类使用您的train_Medical训练,效果很差

Open WithAIGC opened this issue 3 years ago • 4 comments

我想问一下,你的数据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))

`

WithAIGC avatar Dec 27 '21 10:12 WithAIGC

这和你的数据集有关呀。

bubbliiiing avatar Jan 03 '22 15:01 bubbliiiing

我的意思,用自己的数据集,这里需要修改吗modify_png[png <= 127.5] = 1

WithAIGC avatar Jan 04 '22 08:01 WithAIGC

这和你的数据集格式有关………………

bubbliiiing avatar Jan 06 '22 15:01 bubbliiiing

可以解释一下,什么样的格式数据集需要modify_png[png <= 127.5] = 1

WithAIGC avatar Jan 07 '22 01:01 WithAIGC