search

vt-vl-lab / iCAN

[BMVC 2018] iCAN: Instance-Centric Attention Network for Human-Object Interaction Detection
https://gaochen315.github.io/iCAN/
MIT License
258 stars 60 forks source link

How to generate training .pkl file? #7

Closed kaifengcai closed 5 years ago

kaifengcai commented 5 years ago

Hi Chen, When I trained an iCAN on V-COCO, the network read data from .pkl file which was downloaded from google Driver. Could you please tell me how to put the images dataset in a .pkl file? Do you have the code?

Thanks a lot.

gaochen315 commented 5 years ago

For training, I convert the V-COCO and HICO-DET GT annotation into the format I preferred and save it as a pickle file. For testing, I use Detectron to generate object detections and save them as a pickle file. Take a look at misc/Object_Detector.py line 100~111.

Hope this helps.