AttributeError: 'CellposeModel' object has no attribute 'diam_labels'[BUG]

[VerbeekM]

I am training a model using the GUI. When I train the model using my annotated set of images with nuclei as an initial model all is well. However, when I swap the initial model for the scratch model I get the following error without changing the input images. I am using cellpose 3.

LOG:
2024-03-26 17:34:59,427 [INFO] 90, train_loss=0.0459, test_loss=0.0000, LR=0.1000, time 72.43s
Traceback (most recent call last):
File "/.conda/envs/cellposeyml/lib/python3.12/site-packages/cellpose/gui/gui.py", line 1958, in new_model
self.train_model(restore=restore, normalize_params=normalize_params)
File "/.conda/envs/cellposeyml/lib/python3.12/site-packages/cellpose/gui/gui.py", line 1992, in train_model
diam_labels = self.model.diam_labels #.copy()
^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'CellposeModel' object has no attribute 'diam_labels'

Any idea how I could solve this? Thank you!

[tanpancodes]

I had a similar issue before... can you send a picture of the code you used//commands you executed (more log info)

[XylotrupesGideon]

I ran into the same issue when I forgot to use the keyword pretrained_model in cellpose_models_CellposeModel

it occured when i did:
cellpose.models.CellposeModel(cp_pretrained_model)
instread of
cellpose.models.CellposeModel(pretrained_model = cp_pretrained_model)