search

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
2.72k stars 600 forks source link

size problem #13

Closed Nastu-Ho closed 5 years ago

Nastu-Ho commented 5 years ago

Does this model have a requirement for image size?I have uesd 2 medical dataset in this model,but the results are strange. 这个模型对输入图片大小有要求吗？并且我的dataset是单通道图片，我把输入图片调整为三通道，但结果出来异常，有些指标很低

LeeJunHyun commented 5 years ago

Hi @Nastu-Ho ,

This code is implemented for ISIC images. If you have different size of dataset, plz check the data_loader.py. There are some codes for cropping and to resizing images, I think this part affect your results.

Thank you :)

LeeJunHyun commented 5 years ago

If you don't have further issues, I will close the issue. I hope my answer was helpful :)