CASIA-IVA-Lab / FastSAM

Fast Segment Anything
GNU Affero General Public License v3.0
7.34k stars 682 forks source link

Why isn't the model I trained working??? #239

Open xpbag opened 1 month ago

xpbag commented 1 month ago

I use the training code `from ultralytics import YOLO if name == 'main':

model = YOLO(model="FastSAM-x.pt")
model.train(data="aaa.yaml",
            epochs=200,
            batch=32,
            imgsz=1024,
            overlap_mask=False,
            save=True,
            save_period=5,
            device=[0,1],
            project='fastsam',
            name='test',
            val=False)`

Is it normal for the size of the model to double after training??? And when I used the trained model to reason, I still reported an error indicating that ultralytics.util was missing, but when I filled this in the code, it became a case of only running, no output, so i want ask about it why

joshua-atolagbe commented 1 month ago

Ensure your version of ultralytics == 8.0.141

zgf1739 commented 22 hours ago

我使用训练代码 'from ultralytics import YOLO if name == 'main':

model = YOLO(model="FastSAM-x.pt")
model.train(data="aaa.yaml",
            epochs=200,
            batch=32,
            imgsz=1024,
            overlap_mask=False,
            save=True,
            save_period=5,
            device=[0,1],
            project='fastsam',
            name='test',
            val=False)`

训练后模型的大小翻倍是否正常???而当我使用训练好的模型进行推理时,我仍然报告了一个错误,表明 ultralytics.util 丢失了,但是当我在代码中填写这个时,它变成了只运行,没有输出的情况,所以我想问一下为什么

你好请问您的问题解决了么 能不能给个答复 谢谢 Hello, is your problem solved, can you give an answer, thank you