Comments (12)
Hi,
Thank you for your interest. Can you show the full config including the _BASE_
? If you are using the exact CenterNet2_R50_1x, you don't need to copy these configs as they are contained in the base file. If your _BASE_
is already Base-CenterNet2.yaml, can you also provide the command of your training?
Best,
Xingyi
from centernet2.
Yes, my BASE is already Base-CenterNet2.yaml. Trainning command is:
python train_net.py --num-gpus 8 --config-file configs/CenterNet2_R50_1x.yaml
from centernet2.
Hi,
I have shared my training and evaluation log for the R50-1x model here. Please have a check if there is anything clearly mismatched. If it doesn't work, I am happy to check your training log if you can upload it.
from centernet2.
My traing log is very different from yours.
My loss like: total_loss: 1.169 loss_box_reg_stage0: 0.1548 loss_box_reg_stage1: 0.2466 loss_box_reg_stage2: 0.2614 loss_centernet_agn_neg: 0.02222 loss_centernet_agn_pos: 0.05661 loss_centernet_loc: 0.1677 loss_cls_stage0: 0.09623 loss_cls_stage1: 0.08068
But you is: stage0/fast_rcnn/cls_accuracy: 0.944 stage0/fast_rcnn/fg_cls_accuracy: 0.755 stage1/fast_rcnn/cls_accuracy: 0.949 stage1/fast_rcnn/fg_cls_accuracy: 0.780 stage2/fast_rcnn/cls_accuracy: 0.955 stage2/fast_rcnn/fg_cls_accuracy: 0.773 total_loss: 1.401 loss_box_reg_stage0: 0.177 loss_box_reg_stage1: 0.260 loss_box_reg_stage2: 0.262 loss_centernet_agn_neg: 0.033 loss_centernet_agn_pos: 0.087 loss_centernet_loc: 0.198 loss_cls_stage0: 0.144 loss_cls_stage1: 0.126 loss_cls_stage2: 0.115 time: 0.5513 data_time: 0.0178 lr: 0.000200 max_mem: 5000M.
Is it because a different version of detectron2 is used?
you model output config has:
EFFICIENTNET:
BASE_NAME: efficientnet_b3
NORM: FrozenBN
OUT_LEVELS: (3, 4, 5)
why EFFICIENTNET in the config? Your Base-CenterNet2.yaml is this one ?
from centernet2.
Hi,
Sorry for the delayed response. It seems I do not have access to your log, can you share it? For the difference in the log format, yes I used my own detectron2 version which I printed more statistics during training (simply modifying this line). EfficientNet is in my own version and is not used in this project. These should not affect the functionality of the codebase. The referred Base-CenterNet2.yaml is correct. I'll have a better idea when I see your log.
from centernet2.
Hello, where can I download this dataset(coco_un_yolov4_55_0.5)?
Thanks! @xingyizhou
from centernet2.
My traing log.
Another Question: I want to train detect with my self dataset, some boxes have not category, some boxes have category. I use two ways:
1: add another category;
2: didnot add another category, set -999 as the category which box has no category;
box has no category doesn't Calculation the cls-loss. I trained on coco, mask half bbox with no category, validation dataset bbox all have category. Two ways both has 30 MAP, but the loss is lagger when the AP goes on;
total_loss: 10.1 loss_box_reg_stage0: 2.176 loss_box_reg_stage1: 2.417 loss_box_reg_stage2: 4.492 loss_centernet_agn_neg: 0.03299 loss_centernet_agn_pos: 0.1173 loss_centernet_loc: 0.2441 loss_cls_stage0: 0.1453 loss_cls_stage1: 0.1776 loss_cls_stage2: 0.2449
The box regresion loss is very big,but the MAP on validate dataset get better. Is there any suggestions ?
The way 2 trainning log on coco;
from centernet2.
The log shows you are using a batchsize of 96. Can you use the original batch size (16) or modify the total iterations and learning rate accordingly? Please do specify any changes you made in the code when reporting reproducibility issues.
from centernet2.
@WangBoying you can download it here from the model zoo.
from centernet2.
This is very important to me! Thank you very much!
from centernet2.
@xingyizhou The original code and config trained log
from centernet2.
@liuheng0111 Did you found the solution?I have a similar problem with you
from centernet2.
Related Issues (20)
- About Gaussian kernel calculation HOT 1
- GPU memory is not fully used. HOT 7
- run demo.py for video output error HOT 6
- Inference using demo.py issue
- CUDA memory usage continuously increases HOT 3
- Missing config file?
- How about the FLOPs of Centernet2? vs YOLOX, which is better?
- Cannot find models/CenterNet2_R50_1x.pth HOT 1
- TypeError: 'NoneType' object is not iterable HOT 2
- Is it easy to add rotated proposal to centernet2? HOT 1
- Reproduced CenterNet-S4_ DLA_ 8x with RTX3090, but the FPS is only 26.4, far from 71 HOT 3
- hangs while calling comm.synchronize() HOT 1
- About pre-trained model
- Hi, I can not find the config file of CenterNet2_DLA-BiFPN-P5_640_24x_ST.yaml. What should I do? please
- A bout the modelzoo downloading HOT 1
- Cannot download models from model zoo? HOT 1
- What does agn_hm stand for? HOT 2
- Which config corresponds to the centernet * proposed in the paper
- Please read & provide the following
- Where I can get probabilities distributions in the output?
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 centernet2.