cherubicXN / hawp

Holistically-Attracted Wireframe Parsing [TPAMI'23] & [CVPR' 20]
MIT License
291 stars 51 forks source link

Training with own data #50

Open yizhidecainiao opened 1 year ago

yizhidecainiao commented 1 year ago

Hello and thank you for your excellent work. Can you please tell me how I can train my own unlabeled data by pre-training the model and how should I organize the config file.

cherubicXN commented 1 year ago

Hi, please check out [docs/HAWPv3.train.md] for the training details. For your own data, you may need to modify the dataset class.

yizhidecainiao commented 1 year ago

Hi, please check out [docs/HAWPv3.train.md] for the training details. For your own data, you may need to modify the dataset class.

Thanks for your reply!