问题:

训练时在模型结构后只打印以下参数,无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()
Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐