search

cherubicXN / hawp

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

the new pth #54

Closed pipigenius closed 10 months ago

pipigenius commented 10 months ago

Great work. Is hawpv3-ce3ae2cb.pth not available right now? Thank you

cherubicXN commented 10 months ago

Hi, thanks for your kind words. The new checkpoint hawpv3-ce3ae2cb.pth was uploaded last week at https://github.com/cherubicXN/hawp-torchhub/releases/tag/HAWPv3

pipigenius commented 10 months ago

I replaced hawpv3-fdc5487a.pth with the hawpv3-ce3ae2cb.pth, but there is something wrong with _statedict.

cherubicXN commented 10 months ago

Hi @pipigenius, you may need to update the code in the first, and specify the argument --metarch as HAWP-hrheat. Below is an example

python -m hawp.ssl.predict --metarch HAWP-hrheat --cfg hawp/ssl/config/hawpv3-hrheat.yaml  --ckpt checkpoints/hawpv3-ce3ae2cb.pth --img ~/images/*.JPG -t=0.05
pipigenius commented 10 months ago

Fantastic, it performs better! Thank you Xue Nan :)