Pretrained Model Availability

[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.

Thank you so much in advance for your help!