bbepoch / HoiTransformer

This is the code for HOI Transformer
Apache License 2.0
140 stars 21 forks source link

能训练自己的数据集吗 #17

Closed leolv131 closed 3 years ago

leolv131 commented 3 years ago

怎么训练自己的数据集?

bbepoch commented 3 years ago

当然可以,而且很简单, 1、根据readme里的Annotations部分,构造标注。 2、另外,你可以参考数据集datasets下的hoia.py的25-67行来更改自定义的object和verb。

如果还有问题,继续交流~