diff --git a/apps/focused-vision/backend/src/depthai_models/scrfd-person-detection.yaml b/apps/focused-vision/backend/src/depthai_models/scrfd-person-detection.yaml new file mode 100644 index 000000000..c6530afbb --- /dev/null +++ b/apps/focused-vision/backend/src/depthai_models/scrfd-person-detection.yaml @@ -0,0 +1,2 @@ +model: luxonis/scrfd-person-detection:25g-640x640 +platform: RVC4 diff --git a/apps/focused-vision/backend/src/depthai_models/yunet.yaml b/apps/focused-vision/backend/src/depthai_models/yunet.yaml new file mode 100644 index 000000000..458851bed --- /dev/null +++ b/apps/focused-vision/backend/src/depthai_models/yunet.yaml @@ -0,0 +1,2 @@ +model: luxonis/yunet:320x240 +platform: RVC4 diff --git a/apps/focused-vision/backend/src/main.py b/apps/focused-vision/backend/src/main.py index fc6fd9ada..3b9d84fcd 100644 --- a/apps/focused-vision/backend/src/main.py +++ b/apps/focused-vision/backend/src/main.py @@ -7,7 +7,8 @@ from arguments import initialize_argparser from depthai_nodes.node.extended_neural_network import ExtendedNeuralNetwork from depthai_nodes.node.stage_2_neural_network import Stage2NeuralNetwork - +import os +from pathlib import Path logger = logging.getLogger(__name__) @@ -17,8 +18,14 @@ frame_type = dai.ImgFrame.Type.BGR888i HIGH_RES_WIDTH, HIGH_RES_HEIGHT = 2000, 2000 LOW_RES_WIDTH, LOW_RES_HEIGHT = 640, 640 -PEOPLE_DETECTION_MODEL = "luxonis/scrfd-person-detection:25g-640x640" -FACE_DETECTION_MODEL = "luxonis/yunet:320x240" +SCRIPT_DIR = Path(os.path.dirname(os.path.abspath(__file__))) +DEPTHAI_MODELS_DIR = SCRIPT_DIR / "depthai_models" +PEOPLE_DETECTION_MODEL = dai.NNModelDescription.fromYamlFile( + DEPTHAI_MODELS_DIR / "scrfd-person-detection.yaml" +) +FACE_DETECTION_MODEL = dai.NNModelDescription.fromYamlFile( + DEPTHAI_MODELS_DIR / "yunet.yaml" +) if not args.fps_limit: args.fps_limit = 13 diff --git a/apps/focused-vision/oakapp.toml b/apps/focused-vision/oakapp.toml index c7a4d5b8b..d07a526d9 100644 --- a/apps/focused-vision/oakapp.toml +++ b/apps/focused-vision/oakapp.toml @@ -14,6 +14,8 @@ build_steps = [ "chmod +x /etc/service/backend/run", ] +depthai_models = { yaml_path = "./backend/src/depthai_models" } + [static_frontend] dist_path = "./frontend/dist" @@ -28,4 +30,4 @@ oauth_url = "https://auth.docker.io/token" auth_type = "repository" auth_name = "luxonis/oakapp-base" image_name = "luxonis/oakapp-base" -image_tag = "1.2.6" \ No newline at end of file +image_tag = "1.2.6"