文章

Day13 - 删除ultralytics/assets报错

发现了ultralytics的一个bug

调用ultralytics.utils.torch_utilsget_flops函数之后会导致传入的训练参数不可用

1
2
3
4
5
6
7
from ultralytics import YOLO
from ultralytics.utils.torch_utils import get_flops

model=YOLO("./models/yolov8m-sod.yaml")
print(get_flops(model))
model.train(data="./datasets/VisDrone.yaml", epochs=1)

观察输出结果,他的data依然是默认的coco8.yamlepochs也是100,注释掉train这行后会发现他依然会使用默认参数训练一遍,但是看源码看不出来哪里调用了train的相关函数

删除了ultralytics/assets竟然会报错,他竟然强依赖这张图片,无语了

本文由作者按照 CC BY 4.0 进行授权