openseg-group / openseg.pytorch

The official Pytorch implementation of OCNet, OCRNet, and SegFix.
MIT License
1.19k stars 140 forks source link

Pytorch-1.7 checkpoints? #74

Closed adwiii closed 3 years ago

adwiii commented 3 years ago

This commit notes that the supplied checkpoints in the model zoo are trained using 0.4.1 and that 1.7 will be released soon. Is there a timeline to release the checkpoints for Pytorch-1.7?

PkuRainBow commented 3 years ago

@adwiii Thanks for your interest in our work!

We will update the code and support HRNet + Transformer architecture. The related checkpoints based on PaddlePaddle version will also be included then.

Italy2006 commented 2 years ago

@adwiii Thanks for your interest in our work!

We will update the code and support HRNet + Transformer architecture. The related checkpoints based on PaddlePaddle version will also be included then.

But for scripts/cityscapes/hrnet/run_h_48_d_4_ocr.sh, I found that its model is still torch0.4.0,When are you free to update?