Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
909 views
in Technique[技术] by (71.8m points)

computer vision - Strange semantic segmentation results with UNet in Pytorch

I was just trying to train UNet from scratch with a mammography dataset to detect tumor tissue in mammograms. after 35 training the model I got weird curves and bad results on the train and test set. I couldn't find any obvious mistake in my code, and everything seems correct.

Here is the link to implementation on GitHub: GitHub - hrnademi/Mammography

Further information about the dataset: Training images: 2400 images with the size of 256x256 png Test images: 600 images with the size of 256x256 png

Here are the curves:

enter image description here enter image description here enter image description here enter image description here enter image description here enter image description here

Many thanks for your attention.

question from:https://stackoverflow.com/questions/65872009/strange-semantic-segmentation-results-with-unet-in-pytorch

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)
Waitting for answers

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...