Comments (6)
Oh, I implemented only for batch size 1.
At the first time, I had wanted to support more than batch size, but I didn't need that.
Sorry for confusion of code.
I'm not 100% sure, but I think I applied sigmoid function because of thresholding.
(To make binary segmentation results with threshold 0.5)
You can fix that part as you want :)
from image_segmentation.
Hi, @Lijiatu321 .
I am not familiar with ISBI challenge, but is it classification task?
(because you only mentioned accuracy metric)
U_Net architecture is for segmentation, not for classification.
Could you please explain about the task and the dataset?
from image_segmentation.
Hi,@LeeJunHyun Thank you for you reply.I'm sorry I didn't make it clear before. isbi challenge is a segmantation task.In this challenge, a full stack of EM slices will be used to train machine learning algorithms for the purpose of automatic segmentation of neural structures .The original paper(U-net) use the isbi challenge data to train the network.Here is the linkhttp://brainiac2.mit.edu/isbi_challenge/. There are only 30 pictures in the training set and 30 picture in the testing set. The picture and the mask are all gray. Here is the example of the picture and the ground truth.
from image_segmentation.
Thank you for kind explanation.
How about result images? Did you check the results?
from image_segmentation.
Hi, @LeeJunHyun . I have finally find the problem. There are two things.
One thing is that something wrong with the evaluation. Let me take get_accuracy() as an example, when you call the function you can calculate the accuracy of one batch size, but in the solver.train() when a epoch finished you do the acc = acc/length. And the length is the size of data(number of the train picture), in my opinion i think the length should be the length of the train loader. When we use the batch size 1 that is all right but if we change the batch size the accuracy will be decline. I am not sure if i am right . So, would you like to check you code and give me the answer?
The other thing is that the result of the model that i trained is not bad (accuracy : 90 percent), but i have some questions about the output of the network. When training you use the output of the network and the
gt as the input of the evaluate function but during validate you add the sigmod function. i don't understand why.
from image_segmentation.
Thank you very much for your patient and time , wish you have a good day!!
from image_segmentation.
Related Issues (20)
- Attention gate after upsampling HOT 5
- edge pixel weight map
- some queries HOT 1
- How long will it cost for training on DRIVE dataset?
- saving checkpoints in "model.pth" format and visualizing the segmentation results HOT 1
- JS=1,DS>1and loss>1000 HOT 1
- RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 160 and 384 in dimension 2 at /pytorch/aten/src/TH/generic/THTensor.cpp:689 HOT 8
- Grayscale image HOT 1
- about the .pkl file HOT 2
- > hey guys. use pytorch<=1.2.0 (not confirmed) or change the funtion in "evaluation" to fit the calculation of **bool tensor** will sovle the problem HOT 1
- El
- How to train the multi-class task
- Dice coefficience
- the length variable in the solver.py,
- ValueError: num_samples should be a positive integer value, but got num_samples=0 HOT 1
- License
- the test of solver.py HOT 1
- acc error HOT 1
- Environment details HOT 2
- ValueError: num_samples should be a positive integer value, but got num_samples=0 HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from image_segmentation.