51 lines
1.9 KiB
C++
51 lines
1.9 KiB
C++
|
#include "DetectAlgorithm.h"
|
||
|
|
#include "BoostLog.h"
|
||
|
|
#include "ImageUtilities.h"
|
||
|
|
#include <ds_pedestrian_mot_hisi.h>
|
||
|
|
#include <filesystem>
|
||
|
|
#include <thread>
|
||
|
|
|
||
|
|
DetectAlgorithm::DetectAlgorithm() {
|
||
|
|
}
|
||
|
|
|
||
|
|
DetectAlgorithm::~DetectAlgorithm() {
|
||
|
|
if (m_handle != nullptr) {
|
||
|
|
ds_pedestrian_hisi_release(&m_handle);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void DetectAlgorithm::detect(const uint8_t *nv21ImageData, uint64_t frameIndex) {
|
||
|
|
if (m_handle == nullptr) { // 一定得在这里执行
|
||
|
|
initialize();
|
||
|
|
}
|
||
|
|
ImageUtilities::NV21ToBGR24(nv21ImageData, m_rgbImageBuffer.data(), DetectWidth, DetectHeight);
|
||
|
|
|
||
|
|
std::vector<PedestrianRect> pedestrians;
|
||
|
|
std::vector<io_TrackData> tracks;
|
||
|
|
ds_pedestrian_det_hisi(m_handle, m_rgbImageBuffer.data(), pedestrians);
|
||
|
|
ds_pedestrian_track_hisi(m_handle, m_rgbImageBuffer.data(), frameIndex, pedestrians, tracks);
|
||
|
|
LOG(info) << "frame: " << frameIndex << ", pedestrians: " << pedestrians.size() << ", tracks: " << tracks.size();
|
||
|
|
}
|
||
|
|
|
||
|
|
void DetectAlgorithm::initialize() {
|
||
|
|
constexpr auto licensePath = "/kdata/net.lic";
|
||
|
|
bool licenseExisted = std::filesystem::exists(licensePath);
|
||
|
|
if (licenseExisted && std::filesystem::file_size(licensePath) <= 0) {
|
||
|
|
LOG(warning) << "license " << licensePath << " content is empty, remove it.";
|
||
|
|
std::filesystem::remove(licensePath);
|
||
|
|
licenseExisted = false;
|
||
|
|
}
|
||
|
|
ds_pedestrian_hisi_set_lic_path("/kdata");
|
||
|
|
int status = ds_pedestrian_hisi_init(&m_handle, "/system/models/ds_mot_m0_2000.bin",
|
||
|
|
"/system/models/ds_mot_m1_2000.bin", DetectWidth, DetectHeight, 3);
|
||
|
|
if (status != 0) {
|
||
|
|
LOG(error) << "ds_pedestrian_hisi_init() failed, status: " << status;
|
||
|
|
m_handle = nullptr;
|
||
|
|
} else {
|
||
|
|
LOG(info) << "detect algorithm initialization successfully.";
|
||
|
|
}
|
||
|
|
if (!licenseExisted && std::filesystem::exists(licensePath)) {
|
||
|
|
system("sync");
|
||
|
|
}
|
||
|
|
}
|