MaixCDK

#include "maix_nn.hpp"
#include "maix_camera.hpp"
#include "maix_display.hpp"

int main()
{
    nn::Classifier classifier("/root/models/mobilenetv2.mud");
    image::Size input_size = classifier.input_size();
    camera::Camera cam = camera::Camera(input_size.width(), input_size.height(), classifier.input_format());
    display::Display disp = display::Display();

    char msg[64];
    while(!app::need_exit())
    {
        image::Image *img = cam.read();
        auto *result = classifier.classify(*img);
        int max_idx = result->at(0).first;
        float max_score = result->at(0).second;
        snprintf(msg, sizeof(msg), "%5.2f: %s", max_score, classifier.labels[max_idx].c_str());
        img->draw_string(10, 10, msg, image::COLOR_RED);
        disp.show(*img);
        delete result;
        delete img;
    }
    return 0;
}

from maix import camera, display, image, nn

classifier = nn.Classifier(model="/root/models/mobilenetv2.mud")
cam = camera.Camera(classifier.input_width(), classifier.input_height(), classifier.input_format())
disp = display.Display()

while 1:
    img = cam.read()
    res = classifier.classify(img)
    max_idx, max_prob = res[0]
    msg = f"{max_prob:5.2f}: {classifier.labels[max_idx]}"
    img.draw_string(10, 10, msg, image.COLOR_RED)
    disp.show(img)