Closed golnooshAbd closed 2 years ago
Hi @junyuchen245 ,
You need to modify this configuration according to the pre-trained weights: https://github.com/junyuchen245/TransMorph_Transformer_for_Medical_Image_Registration/blob/f89c4e46ae77e17952da5350a86ac95b1ff0351a/IXI/TransMorph/infer_TransMorph.py#L31
If you change this to the line below, then the pre-trained weight should work.
config = CONFIGS_TM['TransMorph']
Each of these strings corresponds to a specific configuration of TransMorph:
https://github.com/junyuchen245/TransMorph_Transformer_for_Medical_Image_Registration/blob/ec422e6b0beb7989340c24659089c3e889ad9b66/IXI/TransMorph/models/TransMorph.py#L883-L894
However, please note that we only provided the pre-trained weights for the base model 'TransMorph'
.
Thanks, Junyu
I was trying to inference pre-trained model weights of TransMorph (simple) but after running infer_TransMorph.py in terminal, I faced these errors:
It seems to be a problem with pre-trained weights, would you please check and help me fix it. Thanks in advanced.