Pretrained weight for pytorch

MrGiovanni / ModelsGenesis

[MICCAI 2019 Young Scientist Award] [MEDIA 2020 Best Paper Award] Models Genesis

Other

729 stars 139 forks source link

Pretrained weight for pytorch #15

Closed zhanghaoyue closed 4 years ago

zhanghaoyue commented 4 years ago

It seems that the pretrained weight .h5 is for Keras. Could you please provide .pth pretrained weight for Pytorch?

Thank you

MrGiovanni commented 4 years ago

Hi @zhanghaoyue

We have provided the implementation of Models Genesis in PyTorch — the pre-trained .pt weights will be released soon. We are currently verifying its performance on multiple target tasks (basically redo all the experiments in PyTorch version) to ensure the released .pt weights offering transferability as good as the pre-trained .h5 weights.

Hopefully, the pre-trained .pt weights would be released before mid-Feb.

Best, Zongwei

zhanghaoyue commented 4 years ago

Thank you! glad to know!