Merge pull request #187 from luxonis/update_openvino

Update OpenVINO version in examples to 2021.4
luxonis · Jul 26, 2021 · ac823bd · ac823bd
2 parents 3df1e13 + 5e7582a
commit ac823bd
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 21 deletions.
diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt
@@ -59,41 +59,41 @@ endfunction()
 
 # NeuralNetwork node, mobilenet example
 hunter_private_data(
-    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/mobilenet-ssd_openvino_2021.2_6shave.blob"
-    SHA1 "f0e14978b3f77a4f93b9f969cd39e58bb7aef490"
-    FILE "mobilenet-ssd_openvino_2021.2_6shave.blob"
+    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/mobilenet-ssd_openvino_2021.4_6shave.blob"
+    SHA1 "4f4506726e3083981064938a0faaf9af6180d2c6"
+    FILE "mobilenet-ssd_openvino_2021.4_6shave.blob"
     LOCATION mobilenet_blob
 )
 
 # YoloV3 resource
 hunter_private_data(
-    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/tiny-yolo-v3_openvino_2021.2_6shave.blob"
-    SHA1 "f0ac263a0d55c374e1892eea21c9b7d1170bde46"
-    FILE "tiny-yolo-v3_openvino_2021.2_6shave.blob"
+    URL "http://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/yolo-v3-tiny-tf_openvino_2021.4_6shave.blob"
+    SHA1 "dedb2d4d96b23e42d15c15e454b8f02eca2713de"
+    FILE "yolo-v3-tiny-tf_openvino_2021.4_6shave.blob"
     LOCATION tiny_yolo_v3_blob
 )
 
 # YoloV4 resource
 hunter_private_data(
-    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/tiny-yolo-v4_openvino_2021.2_6shave.blob"
-    SHA1 "219d949610a5760e62a8458941e1300b81c3fe4a"
-    FILE "tiny-yolo-v4_openvino_2021.2_6shave.blob"
+    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/yolo-v4-tiny-tf_openvino_2021.4_6shave.blob"
+    SHA1 "d8d09b697dac298fe83cf8856740a21b1a61ab89"
+    FILE "yolo-v4-tiny-tf_openvino_2021.4_6shave.blob"
     LOCATION tiny_yolo_v4_blob
 )
 
 # NeuralNetwork node, mobilenet example, 5 shaves
 hunter_private_data(
-    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/mobilenet-ssd_openvino_2021.2_5shave.blob"
-    SHA1 "d715f85e474609cf3f696d7a2e3750804ed6c726"
-    FILE "mobilenet-ssd_openvino_2021.2_5shave.blob"
+    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/mobilenet-ssd_openvino_2021.4_5shave.blob"
+    SHA1 "14ab3ef446b8a649d6694d94c2fc8b798efd74a3"
+    FILE "mobilenet-ssd_openvino_2021.4_5shave.blob"
     LOCATION mobilenet_5shaves_blob
 )
 
 # NeuralNetwork node, mobilenet example, 8 shaves
 hunter_private_data(
-    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/mobilenet-ssd_openvino_2021.2_8shave.blob"
-    SHA1 "3329bb8f3a9c881ef9756d232055f9d6f38aa07b"
-    FILE "mobilenet-ssd_openvino_2021.2_8shave.blob"
+    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/mobilenet-ssd_openvino_2021.4_8shave.blob"
+    SHA1 "7e7ea9a680668f3ca9e8f1d22cc8f3f1a02ac018"
+    FILE "mobilenet-ssd_openvino_2021.4_8shave.blob"
     LOCATION mobilenet_8shaves_blob
 )
 
@@ -129,9 +129,9 @@ hunter_private_data(
 
 # person-detection
 hunter_private_data(
-    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/person-detection-0201_openvino_2021.3_7shave.blob"
-    SHA1 "a8d4b0067deb889ec66b2204573d9ee1e0039c8e"
-    FILE "person-detection-0201_openvino_2021.3_7shave.blob"
+    URL "https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/person-detection-retail-0013_openvino_2021.4_7shave.blob"
+    SHA1 "946d102874b287292cdb5952fa66356c8d36025f"
+    FILE "person-detection-retail-0013_openvino_2021.4_7shave.blob"
     LOCATION person_detection
 )
 

diff --git a/examples/src/object_tracker_video.cpp b/examples/src/object_tracker_video.cpp
@@ -6,7 +6,7 @@
 // Inludes common necessary includes for development using depthai library
 #include "depthai/depthai.hpp"
 
-static const std::vector<std::string> labelMap = {"person", ""};
+static const std::vector<std::string> labelMap = {"", "person"};
 
 static std::atomic<bool> fullFrameTracking{false};
 
@@ -49,7 +49,7 @@ int main(int argc, char** argv) {
     // Properties
     xinFrame->setMaxDataSize(1920 * 1080 * 3);
 
-    manip->initialConfig.setResizeThumbnail(384, 384);
+    manip->initialConfig.setResizeThumbnail(544, 320);
     // manip->initialConfig.setResize(384, 384);
     // manip->initialConfig.setKeepAspectRatio(false); //squash the image to not lose FOV
     // The NN model expects BGR input. By default ImageManip output type would be same as input (gray in this case)
@@ -64,7 +64,7 @@ int main(int argc, char** argv) {
     objectTracker->inputTrackerFrame.setBlocking(true);
     objectTracker->inputDetectionFrame.setBlocking(true);
     objectTracker->inputDetections.setBlocking(true);
-    objectTracker->setDetectionLabelsToTrack({0});  // track only person
+    objectTracker->setDetectionLabelsToTrack({1});  // track only person
     // possible tracking types: ZERO_TERM_COLOR_HISTOGRAM, ZERO_TERM_IMAGELESS
     objectTracker->setTrackerType(dai::TrackerType::ZERO_TERM_COLOR_HISTOGRAM);
     // take the smallest ID when new object is tracked, possible options: SMALLEST_ID, UNIQUE_ID