BangguWu / ECANet

Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks
MIT License
1.26k stars 198 forks source link

Training with custom datasets and labels configuration #33

Closed alancarlosml closed 4 years ago

alancarlosml commented 4 years ago

Hello,

I would like to know about the dataset folder configuration to perform my training on detecting objects.

Something like this:

And about label configuration, for instance: img1.txt class,x,y,w,h

BangguWu commented 4 years ago

For detection or segmentation tasks, you can refer to mmdetection. And customize our ecanet as the backbone.