search

QData / C-Tran

General Multi-label Image Classification with Transformers
MIT License
238 stars 40 forks source link

About applications on custom datasets #10

Closed Breeze-Zero closed 2 years ago

Breeze-Zero commented 2 years ago

Thank you very much for your work, but I am confused about some code of your dataset, especially the coco type. Your code is not regular COCO dataset, and you did not provide relevant code to convert to your format, which made it very difficult for me to migrate to my own dataset.

jacklanchantin commented 2 years ago

We use two versions of the COCO dataset. The general version is the COCO 80 object recognition dataset:

http://images.cocodataset.org/zips/train2014.zip http://images.cocodataset.org/zips/val2014.zip

Breeze-Zero commented 2 years ago

We use two versions of the COCO dataset. The general version is the COCO 80 object recognition dataset:

http://images.cocodataset.org/zips/train2014.zip http://images.cocodataset.org/zips/val2014.zip

Sorry, maybe I'm out of my mind, but the common coco data sets I understand are json markup files, and what I see in your code are ''train_data_name = 'train.data' '' and read it with pickles

jacklanchantin commented 2 years ago

We have a processing file to convert from the json annotations format to the .data format. Send me an email and I can send it to you.