search

aim-uofa / AdelaiDet

AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
https://git.io/AdelaiDet
Other
3.37k stars 647 forks source link

How to train my personal data in BoxInst? #436

Open chenweifu2008 opened 3 years ago

chenweifu2008 commented 3 years ago

Hi! I wanna train my personal data(1 class). can u give me some instructions？should i convert my data into coco format? thx!

tianzhi0549 commented 3 years ago

@chenweifu2008 Our project is based on Detectron2, so you can refer to the documents of Detectron2 for this. https://detectron2.readthedocs.io/en/latest/tutorials/datasets.html

The easiest way should be to convert your data to the COCO format, though.

ameyparanjape commented 2 years ago

Steps I followed: 1) Create bbox annotations for your custom data, convert them into COCO format. 2) Update the config parameters according to your custom data. (eg. NUM_CLASSES) 3) Write a custom script in tools/train_net.py to register your train and test data for Pytorch dataloader 4) Training begins