yolo11经验教训----之一

gaoenyang760525

已于 2024-12-02 11:51:33 修改

阅读量409

点赞数 3

文章标签： yolo

于 2024-12-02 11:50:59 首次发布

本文链接：https://blog.youkuaiyun.com/gaoenyang760525/article/details/144184764

版权

一、格式转换

可以把python中的.pt文件，导出为libtorch识别的格式：

model = YOLO("yolo11n.pt")
model.export(format="torchscript")

二、查看结构

在c++中，我用qt，这样做的：

#include "mainwindow.h"
#include "ui_mainwindow.h"


#undef slots
#include <torch/torch.h>
#include <torch/script.h>
#define slots Q_SLOTS



MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);
}

MainWindow::~MainWindow()
{
    delete ui;
}

void MainWindow::on_pushButton_clicked()
{
    try {
        // 加载模型文件
        torch::jit::script::Module model = torch::jit::load("d:/BaiduNetdiskDownload/ultralytics-main/yolo11n.torchscript");
        model.eval();
        
        // 打印模型结构
        for (const auto& child : model.named_children()) {
            qDebug() << "层名称:" << QString::fromStdString(child.name);
            qDebug() << "层类型:" << QString::fromStdString(child.value.type()->str());
            
            // 打印该层的参数信息
            for (const auto& param : child.value.named_parameters()) {
                qDebug() << "参数名称:" << QString::fromStdString(param.name);
                // 获取并打印形状
                auto sizes = param.value.sizes();
                QString shape = "[";
                for(size_t i = 0; i < sizes.size(); ++i) {
                    shape += QString::number(sizes[i]);
                    if(i < sizes.size() - 1) shape += ", ";
                }
                shape += "]";
                qDebug() << "参数形状:" << shape;
            }
            qDebug() << "------------------------";
        }
        
        qDebug() << "模型加载成功";
    }
    catch (const c10::Error& e) {
        qDebug() << "模型加载失败:" << e.what();
    }
}