Model Zoo

1#!/usr/bin/env python3
2
3import time
4import depthai as dai
5
6# Describe the model I want to download - two options: in program or from yaml file
7
8# Option 1: In program
9# modelDescription = dai.NNModelDescription(model="yolov6-nano", platform="RVC2", ...)
10
11# Option 2: From yaml file
12modelDescription = dai.NNModelDescription.fromYamlFile("./mymodel.yaml")
13
14# If you want, you can store the model description in a yaml file
15modelDescription.saveToYamlFile("./mymodel.yaml")
16
17# Return path to downloaded model - yolov6-nano-r2-288x512.tar.xz for this example
18modelPath = dai.getModelFromZoo(modelDescription, useCached=False, progressFormat="pretty")
19print(f"Model path: {modelPath}")
20
21# Load the model (most of the time it's a NNArchive)
22archive = dai.NNArchive(modelPath)
23
24print(f"Input size of the model {archive.getInputSize()}, name is {archive.getConfig().model.metadata.name}")
25# The arhive can then be used with any of the neural network nodes (NeuralNetwork, DetectionNetwork, SpatialDetectionNetwork)

1#include <iostream>
2
3#include "depthai/depthai.hpp"
4
5int main() {
6    // Option 1: In program
7    // dai::NNModelDescription modelDescription;
8    // modelDescription.model = "yolov6-nano";
9    // modelDescription.platform = "RVC2";
10
11    // Option 2: From yaml file
12    auto modelDescription = dai::NNModelDescription::fromYamlFile("./mymodel.yaml");
13    std::cout << "Model description: " << modelDescription << std::endl;
14
15    // If you want, you can store the model description in a yaml file
16    modelDescription.saveToYamlFile("./mymodel.yaml");
17
18    // Return path to downloaded model - yolov6-nano-r2-288x512.tar.xz for this example
19    auto modelPath = dai::getModelFromZoo(modelDescription, false);  // false means don't use cached model
20    std::cout << "Model path: " << modelPath << std::endl;
21
22    // Load the model (most of the time it's a NNArchive)
23    dai::NNArchive archive(modelPath);
24
25    // Get input size and name of the model
26    auto inputSize = archive.getInputSize();
27    if(inputSize) {
28        std::cout << "Input size of the model: " << inputSize->first << "x" << inputSize->second << std::endl;
29    }
30    std::cout << "Model name is: " << archive.getConfig<dai::nn_archive::v1::Config>().model.metadata.name << std::endl;
31
32    return 0;
33}