src/main.cpp - dev - limb/catedit-aurora

src/main.cpp
52 строки2 KB
Артемий
fix: load models
09 июл 2026, 10:44
09 июл 2026, 10:441af5624
#include <QtQuick>
#include <auroraapp.h>

#include <QDebug>
#include <QFile>
#include <QQmlContext>

#include "imageprocessor.h"
#include "neuralstyleengine.h"
#include "resultimageprovider.h"
#include "segmentationengine.h"

int main(int argc, char *argv[])
{
    QScopedPointer<QGuiApplication> application(Aurora::Application::application(argc, argv));
    application->setOrganizationName(QStringLiteral("ru.poly"));
    application->setApplicationName(QStringLiteral("CatEditAI"));

    QScopedPointer<QQuickView> view(Aurora::Application::createView());

    ResultImageProvider *resultProvider = new ResultImageProvider;
    QScopedPointer<ImageProcessor> imageProcessor(new ImageProcessor(resultProvider));

    const QString segmentationParamPath =
            QStringLiteral("/usr/share/ru.poly.CatEditAI/models/u2netp.ncnn.param");
    const QString segmentationBinPath =
            QStringLiteral("/usr/share/ru.poly.CatEditAI/models/u2netp.ncnn.bin");
    const QString styleParamPath =
            QStringLiteral("/usr/share/ru.poly.CatEditAI/models/mosaic.ncnn.param");
    const QString styleBinPath =
            QStringLiteral("/usr/share/ru.poly.CatEditAI/models/mosaic.ncnn.bin");

    qDebug() << "CatEditAI segmentation param:" << segmentationParamPath
             << "exists:" << QFile::exists(segmentationParamPath);
    qDebug() << "CatEditAI segmentation bin:" << segmentationBinPath
             << "exists:" << QFile::exists(segmentationBinPath);
    qDebug() << "CatEditAI style param:" << styleParamPath
             << "exists:" << QFile::exists(styleParamPath);
    qDebug() << "CatEditAI style bin:" << styleBinPath
             << "exists:" << QFile::exists(styleBinPath);

    SegmentationEngine::instance().setModelPaths(segmentationParamPath, segmentationBinPath);
    NeuralStyleEngine::instance().setModelPaths(styleParamPath, styleBinPath);

    view->engine()->addImageProvider(QStringLiteral("result"), resultProvider);
    view->rootContext()->setContextProperty(QStringLiteral("imageProcessor"),
                                            imageProcessor.data());
    view->setSource(Aurora::Application::pathTo(QStringLiteral("qml/CatEditAI.qml")));
    view->show();

    return application->exec();
}