Comments (5)
Hi,
One thing you could try is increasing the radius on this line here to 4 or even more (to thicken edges in ground truths). This could be done in the case that ground truth edges aren't prominent enough.
A few things you could try are
- Use a warmup training strategy (train a network for a few ~5 epochs, and use it as pre-trained network)
- Increase batch size if you can, or utilize gradient accumulation.
- Play around with learning rate, or follow the paper if you wish (as per the paper, "we set the learning rate, step size, gamma and crop size to 1e − 7 / 5e − 8, 10000 / 20000, 0.1 / 0.2 and 352×352 / 472×472 respectively for SBD and Cityscapes").
- The most important point, you'll need to train for a large number of epochs. I had trained for ~200 and saw some semblance of a decent results. The authors train for 40000 epochs, and I think that this may be the issue in your case.
Hope that I was of some help, and that you can solve the problem.
from casenet.
Hi,
Yes, while I didn't run training on cityscapes, I trained it on a custom dataset. The loss I got was extremely high too, in the best case it was ~20000. I didn't create an evaluation script, so I can't say for sure, but the visualizations gave me a hint whether it was acceptable or not.
Apologies if this wasn't very helpful!
from casenet.
Thank you for your reply.
In my case, the loss didn't decrease at all and the results were not that good. Could you please give a hint which parameters could I tune to improve the output?
from casenet.
@anirudh-chakravarthy hi,it's 40000 iteration,not 40000 epoch。
from casenet.
Hi,
One thing you could try is increasing the radius on this line here to 4 or even more (to thicken edges in ground truths). This could be done in the case that ground truth edges aren't prominent enough.
A few things you could try are
- Use a warmup training strategy (train a network for a few ~5 epochs, and use it as pre-trained network)
- Increase batch size if you can, or utilize gradient accumulation.
- Play around with learning rate, or follow the paper if you wish (as per the paper, "we set the learning rate, step size, gamma and crop size to 1e − 7 / 5e − 8, 10000 / 20000, 0.1 / 0.2 and 352×352 / 472×472 respectively for SBD and Cityscapes").
- The most important point, you'll need to train for a large number of epochs. I had trained for ~200 and saw some semblance of a decent results. The authors train for 40000 epochs, and I think that this may be the issue in your case.
Hope that I was of some help, and that you can solve the problem.
Hello! I trained on my own dataset, the loss was xe+07 but after one epoch it decreased to -1e+06. How could this situation happen? Why did the loss decreased to negative value?
from casenet.
Related Issues (16)
- Performance of this code HOT 2
- would be useful to use CASENet for binary edge detection? HOT 1
- How to run convert_bin_to_hdf5.py ?
- Failing at testing HOT 1
- Can you release the final model parameters on cityscrapes?
- process NYUv2
- Questions about side-outputs.
- Ask for converted numpy pretrained weights HOT 2
- how to get the contour like this? HOT 1
- Where is the dir 'data_aug'? HOT 3
- Pytorch Pretrained Model on Google Drive is broken. HOT 2
- How to preprocess the dataset? HOT 2
- Where can I find the metric F -ODS and AP? HOT 2
- Performance of this code HOT 1
- How can I obtain .h5 files 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 casenet.