Skip to content

Pretrained Model Availability #3

Description

@sunni426

Hi, thank you so much for your wonderful work! We are trying to replicate your model pipeline for team1_bestfitting to predict protein localization in microscopy images. We have run the 3 steps in data_process and are trying to run predict_cell.py. However, we don't see a pretrained model available in the repo, so we are facing some issues in the prediction process. Is there a pretrained model available, or do we have to produce that on our own?

This is the line we're trying to run:
!python predict_cell.py --module cls_inception_v3 --model_name cls_inception_v3_cbam --fold 0 --batch_size 4 --in_channels 4 --num_classes 19 --image_size 128 --model_dir

This is the console log. The created folder '99.00_ema.pth' is empty.
Screen Shot 2023-12-03 at 1 22 51 PM

Thank you so much in advance for your help!

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions