Day13 - 删除ultralytics/assets报错
发现了ultralytics的一个bug
调用ultralytics.utils.torch_utils
的get_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.yaml
,epochs
也是100,注释掉train这行后会发现他依然会使用默认参数训练一遍,但是看源码看不出来哪里调用了train的相关函数
删除了ultralytics/assets
竟然会报错,他竟然强依赖这张图片,无语了
本文由作者按照 CC BY 4.0 进行授权