OpenVINO.hpp

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <exception>
#include <map>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>
 
#include "depthai-shared/common/TensorInfo.hpp"
#include "depthai/utility/Path.hpp"
 
namespace dai {
 
class OpenVINO {
   public:
    enum Version { VERSION_2020_3, VERSION_2020_4, VERSION_2021_1, VERSION_2021_2, VERSION_2021_3, VERSION_2021_4, VERSION_2022_1, VERSION_UNIVERSAL };
 
    struct Blob {
        Blob(std::vector<uint8_t> data);
        Blob(const dai::Path& path);
 
        Version version;
        std::unordered_map<std::string, TensorInfo> networkInputs;
        std::unordered_map<std::string, TensorInfo> networkOutputs;
        uint32_t stageCount = 0;
        uint32_t numShaves = 0;
        uint32_t numSlices = 0;
        std::vector<uint8_t> data;
    };
 
    constexpr static const Version DEFAULT_VERSION = VERSION_2022_1;
 
    static std::vector<Version> getVersions();
 
    static std::string getVersionName(Version version);
 
    static Version parseVersionName(const std::string& versionString);
 
    static std::vector<Version> getBlobSupportedVersions(std::uint32_t majorVersion, std::uint32_t minorVersion);
 
    static Version getBlobLatestSupportedVersion(std::uint32_t majorVersion, std::uint32_t minorVersion);
 
    static Version getBlobVersion(std::uint32_t majorVersion, std::uint32_t minorVersion);
 
    static bool areVersionsBlobCompatible(Version v1, Version v2);
 
   private:
    static const std::map<std::pair<std::uint32_t, std::uint32_t>, Version> blobVersionToOpenvinoGuessMapping;
    static const std::map<std::pair<std::uint32_t, std::uint32_t>, std::vector<Version>> blobVersionToOpenvinoMapping;
};
 
}  // namespace dai