Skip to content

Commit d3663ea

Browse files
committed
use archive for yolo models
1 parent 9aca0f0 commit d3663ea

1 file changed

Lines changed: 7 additions & 22 deletions

File tree

io.github.saeugetier.photobooth.json

Lines changed: 7 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -115,32 +115,17 @@
115115
{
116116
"name": "yolo-models",
117117
"buildsystem": "simple",
118-
"build-options": {
119-
"build-args": [
120-
"--share=network"
121-
]
122-
},
123118
"build-commands": [
124-
"python3 -m venv yolo-venv",
125-
". yolo-venv/bin/activate && pip install --no-cache-dir numpy==1.26.4",
126-
". yolo-venv/bin/activate && pip install --no-cache-dir torch==2.4.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu",
127-
". yolo-venv/bin/activate && pip install --no-cache-dir ultralytics==8.3.168 onnxruntime onnx==1.16.1 onnxslim==0.1.59 ncnn==1.0.20250503 rknn-toolkit2==2.3.2",
128-
". yolo-venv/bin/activate && yolo export model=yolo11n-seg.pt format=onnx",
129-
". yolo-venv/bin/activate && yolo export model=yolo11x-seg.pt format=onnx",
130-
". yolo-venv/bin/activate && yolo export model=yolo11n-seg.pt format=ncnn",
131-
". yolo-venv/bin/activate && yolo export model=yolo11x-seg.pt format=ncnn",
132-
"rm -rf yolo-venv"
133-
],
134-
"post-install": [
135-
"rm *.pt",
136-
"rm *.torchscript",
137-
"mkdir ${FLATPAK_DEST}/share/models",
138-
"mv * ${FLATPAK_DEST}/share/models"
119+
"mkdir -p ${FLATPAK_DEST}/share/models",
120+
"install -Dm644 *.onnx ${FLATPAK_DEST}/share/models/",
121+
"install -Dm644 coco.names ${FLATPAK_DEST}/share/models/",
122+
"for d in *_model; do cp -r \"$d\" ${FLATPAK_DEST}/share/models/; done"
139123
],
140124
"sources": [
141125
{
142-
"type": "file",
143-
"path": "models/coco.names"
126+
"type": "archive",
127+
"url": "https://github.com/saeugetier/photobooth-model-zoo/releases/download/v0.1.0/yolo-models-v0.1.0.tar.gz",
128+
"sha256": "48dd52be9d6ab57ce45e237e0a39fbe99275bd19e30a695df6db29e6b1ab65fb"
144129
}
145130
]
146131
},

0 commit comments

Comments
 (0)