yolov8如何用predict.py,改用自己的权重去测试?
不用命令行,要在哪里改,用自己训练得到的best.pt?
yolov8如何用predict.py,改用自己的权重去测试
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
- 邀请回答
-
1条回答 默认 最新
- IT_service_mesh 2023-03-23 21:08关注
参考GPT和自己的思路:
如果您想使用自己训练得到的best.pt权重进行测试,可以按照以下步骤:
在predict.py文件中找到以下代码:
parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)')
将默认的'yolov5s.pt'改为您自己训练得到的best.pt的路径。
如果您的测试数据集与训练数据集在尺寸上不一致,可能需要更改以下代码中的图像大小:
imgsz = opt.img_size # 模型输入的图像大小,默认为640
需要将图像大小更改为测试数据集图像的大小。
找到以下代码:
with torch.no_grad(): for path, img, im0s, vid_cap in dataset: img = torch.from_numpy(img).to(device) img = img.half() if half else img.float() # FP16模式支持 if img.ndimension() == 3: img = img.unsqueeze(0) # Inference pred = model(img)[0] pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)
将输入图片的位置以及推断结果的位置更改为您自己的测试图片和结果存储位置。
执行predict.py文件进行测试。
注意:在更改代码之前,最好将原始代码备份,以便在需要恢复原始设置时使用。
解决评论 打赏 举报无用 1
悬赏问题
- ¥15 zynq烧写程序到flash后不运行
- ¥15 在ECharts中绘制饼图及环形图(图表)
- ¥15 (标签-51单片机|关键词-#include)
- ¥15 裸板ESD问题的思考与实验
- ¥15 electron+python
- ¥50 远程调试PLC的解决方案的问题
- ¥15 微信收付通中,自动分账的情况下,某一接收方分账失败了系统会自动在发起重新分账吗
- ¥15 宝塔ssl证书一直待待验证
- ¥15 相近语义 不同表达 匹配
- ¥15 [vasp+phonopy]表面模型的间隙扩散系数中尝试频率如何用声子频率计算