问题:
训练时在模型结构后只打印以下参数,无GFlops:
解决方案:
终端输入:pip install thop
重新训练即可
若仅查看模型参数信息,不需要训练,使用以下代码:
from ultralytics import YOLO
if __name__ == '__main__':
# Load a model
model = YOLO(r'ultralytics/cfg/models/v8/yolov8.yaml') # 这里放你的模型文件地址
model.info()