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 on custmer dataset? #10

Closed bitwangdan closed 4 years ago

bitwangdan commented 4 years ago

Thank you for sharing the code. I trained on my own dataset(two object, six verb), but my output value is like this. Is there something wrong with my Settings? image @YueLiao

YueLiao commented 4 years ago

I cannot make sure whether it is wrong only based on this confidence map. You can evaluate the results by modifying the evaluation script for your dataset or visualize the predictions.

bitwangdan commented 4 years ago

@YueLiao i have changed the num_class(two) and the num_verbs(six) in hico.py and opts.py，my final eval result is like this, image Is there something I forgot to change?

YueLiao commented 4 years ago

The result may be wrong. I have trained our method for a dataset with 10 classes, and the mAP is about 70%. You should carefully check the code in hico.py, hico_eval.py, decode.py, hoidet.py, etc, and make sure all right for your own dataset.