Comments (7)
i am no longer using colab i am doing it on local now but I can help you with it you just have to go to train_net.py and add these line from detectron2.data.datasets import register_coco_instances
register_coco_instances("my_dataset_train", {}, "/content/results/yolo2coco/train.json", "/content/results/yolo2coco/train")
register_coco_instances("my_dataset_val", {}, "/content/results/yolo2coco/test.json", "/content/results/yolo2coco/test")
register_coco_instances("my_dataset_test", {}, "/content/results/yolo2coco/test.json", "/content/results/yolo2coco/test")
make changes according to your data and add my_dataset_train and my_dataset_test in config file datasets key as mentioned above
from centernet2.
@gauravlochab Thank you for the quick response
from centernet2.
let me know if you get good results :p
from centernet2.
Hi,
From the error log, it seems the config file is not in the correct format. Did you use TRAIN = ('my_dataset_train',)
instead of TRAIN: ('my_dataset_train',)
?
from centernet2.
thanks for the quick reply yes that was the issue, I am training on my custom data on colab with one gpu. could you tell me the best way to fine-tune this on my data what all hyperparameters i should be looking for while training
21:08:40 iter: 186080 total_loss: 2.043 loss_cls_stage0: 0.2828 loss_box_reg_stage0: 0.02241 loss_cls_stage1: 0.2958 loss_box_reg_stage1: 0.02742 loss_cls_stage2: 0.2986 loss_box_reg_stage2: 0.02829 loss_centernet_loc: 0.6763 loss_centernet_agn_pos: 0.3532 loss_centernet_agn_neg: 0.02419 time: 0.8431 data_time: 0.0036 lr: 0.039999
last iteration looks like this. Thanks again
from centernet2.
@gauravlochab Could you share the notebook you are working on? I'm stuck at the data register.
from centernet2.
https://blog.csdn.net/Carlsummer/article/details/116711708
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.