Skip to content

Commit

Permalink
Use existing downloader system for NNArchive blob
Browse files Browse the repository at this point in the history
  • Loading branch information
Jakob Socan committed Apr 22, 2024
1 parent 2d8661e commit 47f9020
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 13 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -37,3 +37,5 @@ _builds/

# clangd cache
.cache/

/env
13 changes: 0 additions & 13 deletions bindings/python/examples/install_requirements.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,6 @@
import argparse
import re
import platform
import urllib.request
import shutil


convert_default = "empty"
parser = argparse.ArgumentParser()
Expand Down Expand Up @@ -182,16 +179,6 @@ def hasWhitespace(string):
else:
subprocess.check_call(cmd)

nn_archive_filename=f"{examples_dir}/models/yolo-v6-openvino_2022.1_6shave-rvc2.tar.xz"
nn_archive_url="https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/nnarchive/yolo-v6-openvino_2022.1_6shave-rvc2.tar.xz"
nn_archive_req = urllib.request.Request(
nn_archive_url, headers={"User-Agent": "Mozilla/5.0"}
)
with urllib.request.urlopen(nn_archive_req) as response, open(
nn_archive_filename, "wb"
) as out_file:
shutil.copyfileobj(response, out_file)

if requireOpenCv and thisPlatform == "aarch64":
from os import environ
OPENBLAS_CORE_TYPE = environ.get('OPENBLAS_CORE_TYPE')
Expand Down
12 changes: 12 additions & 0 deletions bindings/python/examples/models/nn-archive/model.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
description: >-
nn-archive
task_type: object_attributes
files:
- name: yolo-v6-openvino_2022.1_6shave-rvc2.tar.xz
size: 8522072
sha256: 5e6f280597d58ff669a827b41c12222e63fab10e58ddd8d9b402c19fd21d7130
source: https://artifacts.luxonis.com/artifactory/luxonis-depthai-data-local/network/nnarchive/yolo-v6-openvino_2022.1_6shave-rvc2.tar.xz


framework: dldt
license: https://raw.githubusercontent.com/openvinotoolkit/open_model_zoo/master/LICENSE

0 comments on commit 47f9020

Please sign in to comment.