ljwztc / CLIP-Driven-Universal-Model

[ICCV 2023] CLIP-Driven Universal Model; Rank first in MSD Competition.
Other
560 stars 67 forks source link

Can you release Pre-trained Model #1

Closed puallee closed 1 year ago

puallee commented 1 year ago

Thanks for your code. Can you release the Pre-trained Model? Thanks

ljwztc commented 1 year ago

Thanks for your interest. As mentioned in README, you can download pre-trained model with wget https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/swin_unetr.base_5000ep_f48_lr2e-4_pretrained.pt

potpov commented 1 year ago

Hello, Thank you for your incredible work! The weights you provided are specific for the 3D network. Do you have a set of weights for the version with spatial dimension of 2? Thanks.

ljwztc commented 1 year ago

@potpov Hi potpov, thanks for your interest! Since 3d network is better for 3d data, we only train 3d network in this work. But you can replace the backbone in this work with anyother 2d networks.

kretes commented 1 year ago

Hello, I wanted to clarify one thing. The https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/swin_unetr.base_5000ep_f48_lr2e-4_pretrained.pt pretrained model is the one you've used as the starting point (together with CLIP embeddings) for your training, right? It's not the one that you trained in the end?

Have you released the final model weights as well?

ljwztc commented 1 year ago

Hello, I wanted to clarify one thing. The https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/swin_unetr.base_5000ep_f48_lr2e-4_pretrained.pt pretrained model is the one you've used as the starting point (together with CLIP embeddings) for your training, right? It's not the one that you trained in the end?

Have you released the final model weights as well?

Sure, you are right. We don't release final model weight yet, and plan to release the final model weights after reception

GewelsJI commented 1 year ago

Mark this issue and wait for the final model weights.

MrGiovanni commented 1 year ago

Hello @GewelsJI @kretes @potpov @puallee

We just released the trained checkpoints of universal models. Please take a look at the README.

Best,

Zongwei