Could you tell me how to use your code for original datasets?

4uiiurz1 / pytorch-nested-unet

PyTorch implementation of UNet++ (Nested U-Net).

MIT License

851 stars 202 forks source link

Could you tell me how to use your code for original datasets? #7

Open kirohirahanoshi opened 4 years ago

kirohirahanoshi commented 4 years ago

Could you tell me how to use your code for original datasets? Thanks to your guide, I was able to use your code against 2018 Data Science Bowl dataset So I tried to use the original dataset, but I couldn't. Because there is more than one class to identify, I thought. I tried to change the code, but I couldn't. Please help me.

Thank you.

4uiiurz1 commented 4 years ago

@kirohirahanoshi I add support for multi-class segmentation. Please pull the latest commit and try again. Let me know if you have any problems.

zhouyi-git commented 3 years ago

May I ask the author, I have run the data set and code you mentioned, but my IOU is only 0.8 after running, may I ask what is the problem? The original paper has 0.92