search

mkang315 / ASF-YOLO

[IMAVIS] Official implementation of "ASF-YOLO: A Novel YOLO Model with Attentional Scale Sequence Fusion for Cell Instance Segmentation".
GNU Affero General Public License v3.0
63 stars 3 forks source link

Trained Weights #3

Open alevangel opened 8 months ago

alevangel commented 8 months ago

Where is possible to download the pretrained weights?

mkang315 commented 7 months ago

yolov5l-seg.pt that contains pretrained weights has already been uploaded.

alevangel commented 7 months ago

@mkang315 taking image 9 of the BCC validation dataset as an example, I can't get any predictions for the trsh=0.25. Why this may happen?

This is the code in order to repeat : python segment/predict.py --weights yolov5l-seg.pt --source datasets/BCC/images/val/9.bmp

If I reduce the threshold to 0.0001 instead, I get this result: Cattura

mkang315 commented 7 months ago

Sorry, I misunderstood what you mean. The file yolov5l-seg.pt contain the initial weights that I described in the paper and isn't the pretrained weights of ASF-YOLO. I know you want best.pt of ASF-YOLO that I haven't it for now, but you can train the model on the train set and then test examples.