Comments (6)
Did you try to see what is the output when you test your trained network on the train images itself? If the output in that case is also completely class 0, then something is wrong with your training procedure. Did you set set the NoLabels to 2?
Also check that you are viewing the output image properly because 1.0 or 0.0 out of 255.0 on an image would look black.
from pytorch-deeplab-resnet.
The output from the training images is similarly blank. I've set NoLabels=2, and I've also confirmed the values by stepping through in IPython to generate the output image per evalpyt2.py and running np.unique() and np.where() to see that the values are all zero. This is pretty bizarre and I've been trying to figure this out. My segmentation masks are [0, 255] and I made sure to convert the 255 to 1 for foreground labels as your code expects.
I also requested the VOC augmented dataset from you earlier and tried training on that, running the exact command in the repo (train.py --lr 0.00025 --wtDecay 0.0005 --maxIter 20000 --GTpath <train gt images path here> --IMpath <train images path here> --LISTpath data/list/train_aug.txt
) to train for 20k iterations to try to see if it was an issue with my data and I see the same output. I've attached example images comparing your pretrained model against the model output from my training (mine above, pretrained below):
from pytorch-deeplab-resnet.
What IOU do you get when you use evalpyt2.py
to evaluate your model trained on VOC?
from pytorch-deeplab-resnet.
I'm getting a mean IOU of 0.034486 and using the vanilla repository - though I removed unused imports from the top. The machine and cluster work fine for other training procedures.
from pytorch-deeplab-resnet.
There is some error in what you are doing. Are you sure that you have not modified any part of the train script? Just after you reported lower accuracy, I re-ran my model to check its performance. I followed these steps exactly without any modification -
- I cloned my repository.
- I downloaded data. (the same which has been shared with you).
- I also downloaded
MS_DeepLab_resnet_pretrained_COCO_init.pth
from here - I ran train.py --lr 0.00025 --wtDecay 0.0005 --maxIter 20000 --LISTpath data/list/train_aug.txt, I did not need to mention the image and gt paths because I used the default path which you can see using
python train.py -h
I have never used this or any related repository on the computer on which I am training my model currently.
The model is currently training, but I tested my model for the first saved model using evalpty2.py
and got an mIOU for 0.594(59.4%). This value is expected and will improve to around 72.40% as reported in the readme. Please replicate my steps, I believe that you should get these results as well.
from pytorch-deeplab-resnet.
I did remove unused imports and I corrected the integer division / casting per another issue, but it's definitely possible I modified something I forgot about. I greatly appreciate the time you've spent double-checking.
I needed to reduce the scale upper bound and side length in train.py to make the data fit in the GTX 1080's memory, but I got an IOU of 0.6477 for iteration=5000, which lines up with what you're saying. I'm going to check the training for my own dataset.
from pytorch-deeplab-resnet.
Related Issues (20)
- Why is loss for multiple scales calculated in a strange way? HOT 2
- Arbitrary batch size still unimplemented?
- How do get car mask only? HOT 1
- how does iter_size work? HOT 1
- spatial sizes mismatch HOT 1
- the result? HOT 4
- Why change the image channel order after `cv2.imread`
- No Relu in the ClassifierModule HOT 1
- Frozen the statistics of BN? HOT 5
- The size of the prediction HOT 2
- Bad mIOU tested with provided model HOT 8
- problem of train.py HOT 1
- ASPP or LargeFOV? Should be 76.35%.
- docopt HOT 2
- network stucture issue
- read the ground truth of the pascal voc dataset
- where is the crfs implementation? HOT 1
- About image preprocessing HOT 2
- performance 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 pytorch-deeplab-resnet.