how to train on custmer dataset?

[bitwangdan]

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?

@YueLiao

[YueLiao]

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]

@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,

Is there something I forgot to change?

[YueLiao]

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.