Comments (5)
Hi. It is because the semantic and offset branches are not learned yet, leading to empty proposals. To train from scratch, you should train the semantic and offset branch first, then train the backbone. There are two ways to do that.
1) Set prepare_epochs
in the config file to a value greater than 0. Let's assume num_epochs=500, prepare_epochs=100, then the semantic branch will be trained 100 epochs then the whole network will be trained in 400 epochs. (Noted that you also need to empty pretrained_module
and fixed_module
in the config file.
2) Set semantic_only=True
then train the network. Then validate the network with best semantic checkpoint. Then load the model from best semantic checkpoint and fix the backbone before training the ensuing branches.
from softgroup.
Thank you. I tried the first approach. The best performance I got on scannetv2 is mAP/mAP50/mAP25 = 0.435/0.642/0.772.
Could you reproduce the results when training from scratch on scannetv2?
BTW, I used spconv 2.1.
from softgroup.
I will try to reproduce the results from scratch when available. I think the second approach would be better and more stable. If you are available could you try it and let us know.
from softgroup.
Hi @thangvubk . I have started a training from scratch on my custom dataset with method 2. Now I want to visualize in between result on backbone ntwrk (that I'm training --point-wise prediction network (backbone)). Any way to get results?
from softgroup.
Hi @thangvubk . I have tried the second way, if I want to train the top-down refinement network based on the pretrian model, should I set the optimizer from the scratch. And there exists an error when load optimizer params :
raise ValueError("loaded state dict contains a parameter group "
ValueError: loaded state dict contains a parameter group that doesn't match the size of optimizer's group
from softgroup.
Related Issues (20)
- Install issues HOT 3
- AssertionError: No instance result - results/pred_instance/scene0011_00.txt. HOT 1
- "AssertionError: empty batch" error when training your own dataset HOT 4
- What is the split strategy when working with S3DIS data set? HOT 4
- Explanation of some config parameters HOT 2
- TypeError: forward() takes 6 positional arguments but 9 were given(maybe something wrong with the source codes? HOT 1
- How to obtain voxel labels? HOT 2
- Request code to evaluate mPrec mRec mCov mWcov HOT 1
- Transform_train for custom dataset HOT 1
- Train SoftGroup on different Scenarios HOT 1
- Custom dataset visualization HOT 3
- Why do you need "force_fp32"? HOT 2
- Training batch is empty HOT 2
- Inference without label
- How do I run this network with my own data?
- Volume of instances
- Model reproducibility
- BatchNorm not training
- Understanding the how the visualization process works on instance segmentation
- import error can not import ops from module softgroupe.ops 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 softgroup.