IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
https://detrex.readthedocs.io/en/latest/
Apache License 2.0
1.92k stars 202 forks source link

How to train unregular box? #294

Open Joyako opened 11 months ago

Joyako commented 11 months ago

Hi, my data set labels are unregular, where to modify the data preprocessing code? The format of the label is like this:

截屏2023-08-07 上午10 40 43
rentainhe commented 11 months ago

You can try to register your own data following detectron2's official documentation here: https://detectron2.readthedocs.io/en/latest/tutorials/datasets.html

Then you can try to update the data config to load your own dataset.

Joyako commented 11 months ago

@rentainhe thank you for your reply, does data augmentation need to be modified?

rentainhe commented 11 months ago

@rentainhe thank you for your reply, does data augmentation need to be modified?

Yes, we use detectron2's default data augmentation, but it only works on the normal box, you should double-check this in detectron2's source code.

Joyako commented 11 months ago

@rentainhe thank you for your reply, does data augmentation need to be modified?

Yes, we use detectron2's default data augmentation, but it only works on the normal box, you should double-check this in detectron2's source code.

thanks, I got it

rentainhe commented 11 months ago

@rentainhe thank you for your reply, does data augmentation need to be modified?

Yes, we use detectron2's default data augmentation, but it only works on the normal box, you should double-check this in detectron2's source code.

thanks, I got it

You're welcome, detectron2's documentation has detailed tutorial on how to register your own dataset and augmentations, it may be helpful