FastSAM icon indicating copy to clipboard operation
FastSAM copied to clipboard

training and validation

Open bbetula opened this issue 6 months ago • 1 comments

The training code of this model seems to call YOLO for training. Shouldn't the trained model be a YOLOv8n-seg type model, rather than a FastSAM type model?

see this from ultralytics import YOLO model = YOLO(model="yolov8x-seg.yaml",
) model.train(data="sa.yaml",
epochs=100,
batch=32,
imgsz=1024,
overlap_mask=False,
save=True,
save_period=5,
device='0,1,2,3,4,5,6,7',
project='fastsam',
name='test', val=False,)

bbetula avatar Jul 04 '25 18:07 bbetula

no that is correct as it requires yaml file where as your understanding is because of the folder structure having pt

harshnandwana avatar Jul 07 '25 06:07 harshnandwana