Skip to content

Issue finding model #2

@matthieufaillot

Description

@matthieufaillot

Hi, Thank you for making your great software available
I run into problems when trying to get thin slices from thick slices of Head CT data
I installed KevinSR in a conda python 3.6 env on Ubuntu 22.04
I cloned the repo in Pycharms and run the code on CPU
When running "thins_slices = SOUP_GAN(thicks_ori, Z_FAC, 1)" I get :

 Traceback (most recent call last):
  File "/home/dell/PycharmProjects/SOUP-GAN/Example_MF.py", line 25, in <module>
    thins_slices = SOUP_GAN(thicks_ori, Z_FAC, 1)
TypeError: SOUP_GAN() takes 2 positional arguments but 3 were given

When running thins_slices = SOUP_GAN(thicks_ori, Z_FAC) I get

File "/home/dell/PycharmProjects/SOUP-GAN/Example_MF.py", line 25, in
thins_slices = SOUP_GAN(thicks_ori, Z_FAC)
File "/home/dell/anaconda3/envs/KevinSR/lib/python3.6/site-packages/KevinSR/SOUP.py", line 21, in SOUP_GAN
new_model=keras.models.load_model('./Thin-to-thin')
File "/home/dell/anaconda3/envs/KevinSR/lib/python3.6/site-packages/keras/saving/save.py", line 205, in load_model
return saved_model_load.load(filepath, compile, options)
File "/home/dell/anaconda3/envs/KevinSR/lib/python3.6/site-packages/keras/saving/saved_model/load.py", line 108, in load
meta_graph_def = tf.internal.saved_model.parse_saved_model(path).meta_graphs[0]
File "/home/dell/anaconda3/envs/KevinSR/lib/python3.6/site-packages/tensorflow/python/saved_model/loader_impl.py", line 121, in parse_saved_model
constants.SAVED_MODEL_FILENAME_PB))
OSError: SavedModel file does not exist at: ./Thin-to-thin/{saved_model.pbtxt|saved_model.pb}

Process finished with exit code 1

This is the definition of the SOUP_GAN function in the SOUP.py file :

def SOUP_GAN(thicks_ori, Z_FAC):

thicks_ori = rescale_img(thicks_ori, max_val= 10000)

thins = zoom(thicks_ori, (1,1,Z_FAC))

new_model=keras.models.load_model('./Thin-to-thin')
thins_gen = thins.copy()

target = np.moveaxis(thins,-1,0)
target = target [...,np.newaxis]
target = target [np.newaxis,...]

index  = attention_coeff(target, Z_FAC)

target = new_model.predict([target,index])
 
target_small = target[0,...,0]
thins_gen = np.moveaxis(target_small, 0,-1)

return thins_gen

My understanding is that the SOUP_GAN function
-only takes two arguments ( not three as in example scripts)
-a Thin-to-Thin model by default and that this model is not installed with KevinSR

Is there any way to solve this ? Should I train a model on my own data ?

Best regards
Matthieu

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions