pipi

@timy90022 I am sorry I also neet it to train my own dataset

@wudi00 hi have you solve the problem? @princeton-vl HI, could you tell me how to solve this problem?