Ouwen / MimickNet

Matching clinical-grade ultrasound post-processing without the hassle.
https://arxiv.org/abs/1908.05782
Apache License 2.0
59 stars 14 forks source link

how to load the "padded" version of MimickNet model ? #21

Open Shengwuei opened 1 year ago

Shengwuei commented 1 year ago

Hi Sir,

In Python, below codes load the model "mimicknet_1568473738-210304.h5" successfully

_mimicknet_model = tf.keras.models.load_model('./mimicknet1568473738-210304.h5' compile=False)

However, if change the model file to 'padded_mimicknet_1568473738-210304.h5', it gives me the error :

ValueError: bad marshal data (unknown type code)

After some googling, that might be related to the Python version used when compiling the model file, though I believe you compiled these models under the same Python version (?), so for now I have no idea about how to proceed.

As my understanding, the "padded" version has built in the custom_pad() and custom_depad() function in layer input and output, so I can directly use the model without the codes to define custom_pad() and custom_depad(), so I would prefer to use the "padded" version, thanks and very appreciated for your help .

Ouwen commented 1 year ago

Since colab and tensorflow have updated, the padded version needs to be updated. For now, the colab notebook should be updated to work with latest. However, the padded model will require an older version of tensorflow

Shengwuei commented 1 year ago

Got it, thanks !