search

YueLiao / PPDM

Code for "PPDM: Parallel Point Detection and Matching for Real-time Human-Object Interaction Detection".
MIT License
219 stars 42 forks source link

How to train my own dataset？ #31

Closed 15966697671 closed 3 years ago

15966697671 commented 3 years ago

I want to perform transfer learning to train the model with the pretrained weights on my own small image dataset. My small dataset category is completely different from that of Hoi. Let me know how to go about it.thank you.

YueLiao commented 3 years ago

Sorry for the latter response. First, you should build up a data loader to read and process your data. Second, you modify the output channel of the last convolutional layer of ct_head to the number of categories.