Buckets:
| """cwm3/baselines_dms/registry_seed.py — the CANONICAL provenance record of the DMS-baseline components (2026-09-14 install). | |
| Running it (re)writes weights/registry.json from the entries below, keeping an existing `installed_utc`, and verifies every entry | |
| against the files on disk / the installed distributions (sha256, size, version). Expected hashes were taken from the actual | |
| downloads of 2026-09-14 (registry.verify recomputes them, so a transcription error here surfaces as FAIL, never silently). | |
| python -m baselines_dms.registry_seed | |
| Nothing here reads an openpilot DMS field; every source is public; no API key is involved. | |
| """ | |
| from __future__ import annotations | |
| import sys | |
| from pathlib import Path | |
| from baselines_dms import registry as R | |
| OMZ_REL = "2023.0" | |
| OMZ_BASE = f"https://storage.openvinotoolkit.org/repositories/open_model_zoo/{OMZ_REL}/models_bin/1" | |
| OMZ_LIC = "Apache-2.0" | |
| OMZ_LIC_URL = "https://github.com/openvinotoolkit/open_model_zoo/blob/master/LICENSE" | |
| OMZ_NOTE = (f"Intel Open Model Zoo pre-built FP16 IR (IR v11) from release directory {OMZ_REL} (newest release on storage.openvinotoolkit.org " | |
| "with models_bin/1/<model>/FP16/; 2023.3 / 2024.0 dirs carry no models_bin). Smoke: openvino.Core().compile_model(CPU) OK.") | |
| OF3_COMMIT = "662a555a8566ae3aec8139cb8c72acf6d06e0eb3" | |
| OF3_LIC = "OpenFace 3.0 Software License Agreement — academic / non-profit NONCOMMERCIAL research use only (CMU); RetinaFace part MIT" | |
| OF3_LIC_URL = f"https://github.com/CMU-MultiComp-Lab/OpenFace-3.0/blob/{OF3_COMMIT}/LICENSE" | |
| BLENDSHAPES = ("_neutral,browDownLeft,browDownRight,browInnerUp,browOuterUpLeft,browOuterUpRight,cheekPuff,cheekSquintLeft,cheekSquintRight," | |
| "eyeBlinkLeft,eyeBlinkRight,eyeLookDownLeft,eyeLookDownRight,eyeLookInLeft,eyeLookInRight,eyeLookOutLeft,eyeLookOutRight," | |
| "eyeLookUpLeft,eyeLookUpRight,eyeSquintLeft,eyeSquintRight,eyeWideLeft,eyeWideRight,jawForward,jawLeft,jawOpen,jawRight," | |
| "mouthClose,mouthDimpleLeft,mouthDimpleRight,mouthFrownLeft,mouthFrownRight,mouthFunnel,mouthLeft,mouthLowerDownLeft," | |
| "mouthLowerDownRight,mouthPressLeft,mouthPressRight,mouthPucker,mouthRight,mouthRollLower,mouthRollUpper,mouthShrugLower," | |
| "mouthShrugUpper,mouthSmileLeft,mouthSmileRight,mouthStretchLeft,mouthStretchRight,mouthUpperUpLeft,mouthUpperUpRight," | |
| "noseSneerLeft,noseSneerRight") | |
| def omz(model: str, xml: tuple[int, str], binf: tuple[int, str], io: str, subdir: str | None = None) -> dict: | |
| sub = f"{subdir}/{model}" if subdir else model | |
| files = [{"name": f"{model}.xml", "url": f"{OMZ_BASE}/{sub}/FP16/{model}.xml", "size_bytes": xml[0], "sha256": xml[1]}, | |
| {"name": f"{model}.bin", "url": f"{OMZ_BASE}/{sub}/FP16/{model}.bin", "size_bytes": binf[0], "sha256": binf[1]}] | |
| return dict(id=f"omz_{model.replace('-', '_')}", kind="url", name=model, version=f"OMZ {OMZ_REL} / models_bin/1 / FP16", | |
| url=f"{OMZ_BASE}/{sub}/FP16/", local_path=f"baselines_dms/weights/omz_{model.replace('-', '_')}", | |
| sha256=R.manifest_sha256(files), size_bytes=sum(f["size_bytes"] for f in files), license=OMZ_LIC, license_url=OMZ_LIC_URL, | |
| files=files, notes=f"{OMZ_NOTE} {io}") | |
| def pip(name: str, version: str, wheel_sha: str | None, license: str, license_url: str, notes: str, url: str | None = None, **src) -> dict: | |
| return dict(id=f"pip_{name.replace('-', '_').lower()}", kind="pip", name=name, version=version, url=url or f"https://pypi.org/project/{name}/{version}/", | |
| local_path=None, sha256=wheel_sha, size_bytes=None, license=license, license_url=license_url, notes=notes, source=src or {}) | |
| def build() -> list[dict]: | |
| E: list[dict] = [] | |
| # ---------------------------------------------------------------------------------------------- Intel OpenVINO + OMZ (I1, I2) | |
| E.append(pip("openvino", "2026.3.1", "40bf3724ec70bce0fce4f9547cd1f19a576ef299172ce4ce2552249d10064acc", "Apache-2.0", | |
| "https://github.com/openvinotoolkit/openvino/blob/master/LICENSE", | |
| "pip install openvino (2026-09-14) -> openvino-2026.3.1-22476-cp312-cp312-manylinux_2_28_x86_64.whl; runtime string " | |
| "2026.3.1-22476-759c5a6ab8c-releases/2026/3; devices ['CPU']; also pulled openvino-telemetry 2025.2.0.")) | |
| E.append(pip("openvino-telemetry", "2025.2.0", "bcb667e83a44f202ecf4cfa49281715c6d7e21499daec04ff853b7f964833599", "Apache-2.0", | |
| "https://github.com/openvinotoolkit/openvino_telemetry/blob/master/LICENSE", "dependency of openvino; opt-out via `opt_in_out --opt_out`.")) | |
| E.append(omz("face-detection-adas-0001", (311654, "59ccfd224324a5417210ce863909ae265c89548b19a85364470935201c5176ef"), | |
| (2106088, "10d3d424905ddef043b16fa6c036b7aa14c5193ae2c470699799746d7d6def37"), | |
| "IO: input data [1,3,384,672] f32 BGR; output detection_out [1,1,200,7] (image_id,label,conf,x_min,y_min,x_max,y_max).")) | |
| E.append(omz("head-pose-estimation-adas-0001", (78824, "7a00509b02c15371c8156b848daf2825a061488b6f1b257ccbb648596c0a1ef3"), | |
| (3823816, "035c41b04268c4541fc0d7fc4a131070bc2d424324ad8e1163f83953ee8bb0b1"), | |
| "IO: input data [1,3,60,60] f32 face crop; outputs angle_y_fc [1,1] yaw deg, angle_p_fc [1,1] pitch deg, angle_r_fc [1,1] roll deg (Tait-Bryan).")) | |
| E.append(omz("facial-landmarks-35-adas-0002", (361619, "b6158ac7d58f45b948aa21bcf36271092cf0ae38040b29d269ea0df1f69684f6"), | |
| (9190590, "f75558439fa2a9a183174ed3f43eeaa8eb61024aac1aab907813cb40cc9823b9"), | |
| "IO: input data [1,3,60,60] f32 face crop; output align_fc3 [1,70] = 35 (x,y) normalised landmarks (eyes 0-3, nose 4-7, mouth 8-11, brows 12-17, contour 18-34).")) | |
| E.append(omz("gaze-estimation-adas-0002", (99255, "bc94af22a5185271fbb4b66fd26b6eadead6a2c07f40e063fdeb7bcb22deb84d"), | |
| (3764742, "780f9b184f91967c0fb6a925d6f7b8841931d52e44cfab60e3cce3560072b2d6"), | |
| "IO: inputs left_eye_image [1,3,60,60], right_eye_image [1,3,60,60], head pose angles input named '138' [1,3] (yaw,pitch,roll deg); output gaze_vector [1,3] Cartesian gaze direction (not normalised).")) | |
| E.append(omz("driver-action-recognition-adas-0002-encoder", (194660, "ded6f3e5f4d36c9e74f10fb232e72be3464d37eb16bd1f40e8e0c1726178aef7"), | |
| (5725382, "b340c3da83675936272f37fc6b913e9bbe90f9abce304eff63c4248c1aee1dad"), | |
| "IO: input '0' [1,3,224,224] f32 RGB frame; output '513' [1,512,1,1] per-frame embedding (MobileNetV2 encoder).", subdir="driver-action-recognition-adas-0002")) | |
| E.append(omz("driver-action-recognition-adas-0002-decoder", (234762, "48eefbdbffbfcb804107e3e12e94faeaecd0e3d55c4a052d8253d571f1c652e1"), | |
| (14718156, "92c08eb4b1599694413a2b80e31bdb2b8cb0d90c364a1931d88e99d2026685ee"), | |
| "IO: input '0' [1,16,512] = 16 encoder embeddings; output '804' [1,9] logits over 9 driver actions (Safe driving, Texting left, Talking on the phone left, Texting right, Talking on the phone right, Operating the radio, Drinking eating, Reaching behind, Hair and makeup).", subdir="driver-action-recognition-adas-0002")) | |
| oce_files = [ | |
| {"name": "src/open_closed_eye.onnx", "url": "https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/open-closed-eye-0001/open_closed_eye.onnx", | |
| "size_bytes": 46164, "sha256": "4daa100034482525a26c9afb9297c16580a531189e66e3d2b2ac7d32becfd593"}, | |
| {"name": "src/LICENSE", "url": "https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/open-closed-eye-0001/LICENSE", | |
| "size_bytes": 10254, "sha256": "db231b35ec46df91df83044d2a5c376757833423911459743353f54ddb783c2e"}, | |
| {"name": "src/source.txt", "url": "https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/open-closed-eye-0001/source.txt", | |
| "size_bytes": 105, "sha256": "d2bc8acef2cb97053b78c5966ede6a2707692697b3ad2919cfdae3a9425bc1e1"}, | |
| {"name": "open-closed-eye-0001.xml", "url": None, "size_bytes": 18628, "sha256": "ce2ec244a4479a23d8b379a0b7508a6021c3ec88a320f8f7e5c42c90114d2199"}, | |
| {"name": "open-closed-eye-0001.bin", "url": None, "size_bytes": 22514, "sha256": "4f9eefb146a265e2ec02191eccb7e3e71b7f04c18293181ec35afaeae423ec60"}, | |
| ] | |
| E.append(dict(id="omz_open_closed_eye_0001", kind="url", name="open-closed-eye-0001", version="OMZ public/2022.1 ONNX -> FP16 IR converted locally (openvino 2026.3.1)", | |
| url="https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/open-closed-eye-0001/", local_path="baselines_dms/weights/omz_open_closed_eye_0001", | |
| sha256=R.manifest_sha256(oce_files), size_bytes=sum(f["size_bytes"] for f in oce_files), license="Apache-2.0 (LICENSE shipped next to the ONNX; source: openvino_training_extensions)", | |
| license_url="https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/open-closed-eye-0001/LICENSE", files=oce_files, | |
| notes="OMZ *public* model: no pre-built IR exists in any release directory (0 FP16 dirs in filetree.json). ONNX sha384 matches model.yml checksum " | |
| "2615bce5…cba27 (OMZ master 6697dead). Converted by baselines_dms/omz_convert.py with the model.yml preprocessing (mean 127 / scale 255, output node 19); " | |
| "IO: input input.1 [1,3,32,32] f32 eye crop 0..255; output '19' [1,2,1,1] softmax (0 = open, 1 = closed). Original source per source.txt: download.01.org (offline).")) | |
| # ---------------------------------------------------------------------------------------------- Google MediaPipe (G1) | |
| E.append(pip("mediapipe", "0.10.35", "db9a579df48cffe9570cd3e93f6a5d2dd089a1103b846c60c5b5de8a21c38db0", "Apache-2.0", | |
| "https://github.com/google-ai-edge/mediapipe/blob/master/LICENSE", "pre-existing in hci env (not installed by this task); Tasks API vision.FaceLandmarker.")) | |
| E.append(dict(id="mediapipe_face_landmarker", kind="url", name="face_landmarker.task", | |
| version="float16/latest — GCS generation 1683136941468629, Last-Modified 2023-05-03T18:02:21Z, ETag b0e7274907a1644404fef66b28dd6d85", | |
| url="https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/latest/face_landmarker.task", | |
| local_path="baselines_dms/weights/mediapipe_face_landmarker/face_landmarker.task", | |
| sha256="64184e229b263107bc2b804c6625db1341ff2bb731874b0bcc2fe6544e0bc9ff", size_bytes=3758596, license="Apache-2.0", | |
| license_url="https://ai.google.dev/edge/mediapipe/solutions/vision/face_landmarker#models", | |
| notes="Bundle = BlazeFace short-range detector + face mesh (478 landmarks x,y,z normalised) + blendshape model (52 scores) + facial_transformation_matrixes (4x4). " | |
| "Smoke (mediapipe 0.10.35, output_face_blendshapes=True, output_facial_transformation_matrixes=True): synthetic 256x256 noise -> 0 faces (runs, empty lists); " | |
| "public sample portrait -> 1 face, 478 landmarks, 52 blendshapes, 4x4 matrix, 15 ms CPU. Blendshape category names: " + BLENDSHAPES)) | |
| # ---------------------------------------------------------------------------------------------- L2CS-Net (O3) | |
| E.append(dict(id="l2cs", kind="hf", name="L2CSNet_gaze360.pkl", version="Gaze360 ResNet50 checkpoint (upstream README 'Pretrained models'); mirror revision 424a499580d1217b57f8f1f3c2a5582c30cd6610", | |
| url="https://huggingface.co/tianfxc/l2cs/resolve/424a499580d1217b57f8f1f3c2a5582c30cd6610/L2CSNet_gaze360.pkl", | |
| local_path="baselines_dms/weights/l2cs/L2CSNet_gaze360.pkl", sha256="8a7f3480d868dd48261e1d59f915b0ef0bb33ea12ea00938fb2168f212080665", size_bytes=95849977, | |
| license="MIT (L2CS-Net repository)", license_url="https://github.com/Ahmednull/L2CS-Net/blob/a4d8f7fa5436a2b2b9f088471623b552a85811bd/LICENSE", | |
| source={"repo_id": "tianfxc/l2cs", "revision": "424a499580d1217b57f8f1f3c2a5582c30cd6610", | |
| "official_url": "https://drive.google.com/drive/folders/17p6ORr-JQJcw-eYtG2WGNiuS_qVKwdWd?usp=sharing", | |
| "cross_check": {"repo_id": "py-feat/l2cs", "revision": "7325960c70a6790e4194103e6e30ff83471a2bee", "filename": "l2cs_gaze360_resnet50.safetensors", | |
| "sha256": "75405bdc01f7086b3887280fe50c28a14bc62463a53fa749781e4a3cf98eca0e", "result": "324/324 tensors identical (max |diff| = 0)"}}, | |
| notes="OFFICIAL Google Drive folder from the README (Ahmednull/L2CS-Net@a4d8f7fa) returned HTTP 404 on 2026-09-14 for both the browser URL and `gdown --folder` " | |
| "('Failed to retrieve folder contents … status code 404') -> owner action item 1 (confirm/re-share the official link). Fallback used: unofficial HF re-host " | |
| "tianfxc/l2cs (same file name/size class as upstream), independently cross-checked tensor-by-tensor against the py-feat re-host (documented conversion of the upstream pkl): identical. " | |
| "Smoke: l2cs.Pipeline(weights, arch='ResNet50', device=cuda:0) loads (23.88 M params); synthetic 224x224 -> pitch/yaw arrays (1,) in radians; public portrait -> " | |
| "GazeResultContainer fields pitch, yaw, bboxes (n,4), landmarks (n,5,2), scores (n,).")) | |
| E.append(dict(id="git_l2cs", kind="git", name="l2cs", version="4a0f978d5b4c426a7d37022d8c927d6ea031dcb6", url="https://github.com/edavalosanaya/L2CS-Net.git", | |
| local_path=None, sha256=None, size_bytes=None, license="MIT", license_url="https://github.com/Ahmednull/L2CS-Net/blob/a4d8f7fa5436a2b2b9f088471623b552a85811bd/LICENSE", | |
| source={"pip_args": []}, notes="pre-existing in hci env (pip-installed from the edavalosanaya packaging fork of Ahmednull/L2CS-Net; dist version 0.0.1). Provides l2cs.Pipeline / getArch / GazeResultContainer.")) | |
| E.append(dict(id="git_face_detection", kind="git", name="face-detection", version="786fbab7095623c348e251f1f0a8b323721c6a84", url="https://github.com/elliottzheng/face-detection.git", | |
| local_path=None, sha256="2979b33ffafda5d74b6948cd7a5b9a7a62f62b949cef24e95fd15d2883a65220", size_bytes=1789735, license="MIT", | |
| license_url="https://github.com/elliottzheng/face-detection/blob/786fbab7095623c348e251f1f0a8b323721c6a84/LICENSE", source={"pip_args": []}, | |
| notes="pre-existing (dist 1.0.5); RetinaFace detector used by l2cs.Pipeline; bundles weights/mobilenet0.25_Final.pth (biubug6/Pytorch_Retinaface, sha256 recorded in `sha256`) — " | |
| "byte-identical to OpenFace 3.0's Alignment_RetinaFace.pth.")) | |
| # ---------------------------------------------------------------------------------------------- VideoMAEv2 distilled ViT-S (O4) | |
| E.append(dict(id="videomaev2", kind="hf", name="vit_s_k710_dl_from_giant.pth", version="OpenGVLab/VideoMAE2 revision 706cc172d65ebd4dedbee3f9c0183a93df9fa125 (distill/); model code pinned at VideoMAEv2@29eab1e8a588d1b3ec0cdec7b03a86cca491b74b", | |
| url="https://huggingface.co/OpenGVLab/VideoMAE2/resolve/main/distill/vit_s_k710_dl_from_giant.pth", | |
| local_path="baselines_dms/weights/videomaev2/vit_s_k710_dl_from_giant.pth", sha256="24fb71687fa3671b8387cadfbcbab0f72af695692e93cf1ecc82caa888626172", size_bytes=44334609, | |
| license="Weights: apache-2.0 (HF model-card tag); code: MIT (VideoMAEv2 repo LICENSE)", license_url="https://huggingface.co/OpenGVLab/VideoMAE2 ; https://github.com/OpenGVLab/VideoMAEv2/blob/29eab1e8a588d1b3ec0cdec7b03a86cca491b74b/LICENSE", | |
| source={"repo_id": "OpenGVLab/VideoMAE2", "revision": "706cc172d65ebd4dedbee3f9c0183a93df9fa125", "hf_filename": "distill/vit_s_k710_dl_from_giant.pth", | |
| "vendored_model_file": "baselines_dms/vendor/videomaev2_model.py", "upstream_file": "models/modeling_finetune.py", | |
| "upstream_file_sha256": "d132c97369f25207e380e78c59294fd00b93d2f6aa0fb5c4ded19eff26a5a3cb"}, | |
| notes="Direct download from MODEL_ZOO.md 'Distillation' section — NO request form (the form gates only the giant pre-train/fine-tune weights). Checkpoint dict key 'module' " | |
| "(162 tensors): ViT-S/16, tubelet 2, 16 frames, embed 384, depth 12, heads 6, fc_norm mean-pool, head 710 (K710). Vendored model definition " | |
| "baselines_dms/vendor/videomaev2_model.py = upstream modeling_finetune.py with only the two timm import lines adapted (see header). Smoke: load_state_dict strict " | |
| "<All keys matched>; 1x3x16x224x224 fp16 CUDA forward_features -> (1, 384) finite, 22.15 M params. Used FROZEN (no fine-tuning).")) | |
| # ---------------------------------------------------------------------------------------------- OpenFace 3.0 (O2) | |
| of3_files = [ | |
| {"name": "Alignment_RetinaFace.pth", "url": "https://drive.google.com/uc?id=1IG95GU2i8Yk_asBqQS0MfbJbduI3Fw-T", "size_bytes": 1789735, "sha256": "2979b33ffafda5d74b6948cd7a5b9a7a62f62b949cef24e95fd15d2883a65220"}, | |
| {"name": "Landmark_68.pkl", "url": "https://drive.google.com/uc?id=1wKY6SG_NfJDUksu77w5UwJT9XRwTNrq7", "size_bytes": 176735623, "sha256": "5df84c820c9f8155ec1174252b4361bb9991271e4ffe6d0a203344eaaf95f16e"}, | |
| {"name": "Landmark_98.pkl", "url": "https://drive.google.com/uc?id=1Fn2v2Zde9x4L7ClI0CRxcFP520B-32t7", "size_bytes": 178030153, "sha256": "b49eb62de453fbc3ef278c52b8f85630b536d0a43b82a92d6ff7a11485f12c2a"}, | |
| {"name": "MTL_backbone.pth", "url": "https://drive.google.com/uc?id=12HLZCnp2RHeXZLAIsOvY8QLlKmlU6rT8", "size_bytes": 101710914, "sha256": "72f45843c7a3ed9500835c507467f32e864a23971160f733aeb23797a05a5bcd"}, | |
| {"name": "mobilenetV1X0.25_pretrain.tar", "url": f"https://raw.githubusercontent.com/CMU-MultiComp-Lab/OpenFace-3.0/{OF3_COMMIT}/weights/mobilenetV1X0.25_pretrain.tar", | |
| "size_bytes": 3827150, "sha256": "75c37f8d438b7c6aed2e6b310fd0540c40dabf0dd820770bd7082668dfd7c1c1"}, | |
| ] | |
| E.append(dict(id="openface3", kind="gdown", name="OpenFace 3.0 weights", version=f"Google Drive folder 1aBEol-zG_blHSavKFVBH9dzc9U9eJ92p (files dated 2025-02-23) == HF nutPace/openface_weights@3844412b54706ed0a930b99589508ed3c101f39e; repo {OF3_COMMIT[:12]}", | |
| url="https://drive.google.com/drive/folders/1aBEol-zG_blHSavKFVBH9dzc9U9eJ92p", local_path="baselines_dms/weights/openface3", | |
| sha256=R.manifest_sha256(of3_files), size_bytes=sum(f["size_bytes"] for f in of3_files), license=OF3_LIC, license_url=OF3_LIC_URL, files=of3_files, | |
| source={"folder_id": "1aBEol-zG_blHSavKFVBH9dzc9U9eJ92p", "hf_mirror": "nutPace/openface_weights@3844412b54706ed0a930b99589508ed3c101f39e"}, | |
| notes="`gdown --folder` succeeded (4 files, original names kept). All four sha256 match the HF mirror listed in the README byte-for-byte. The Drive folder lacks " | |
| "mobilenetV1X0.25_pretrain.tar (RetinaFace backbone init that openface-test's RetinaFace(cfg, phase='test') loads from ./weights/ when cfg['pretrain'] is True); " | |
| "fetched from the GitHub repo at the pinned commit (same sha256 as on HF). Alignment_RetinaFace.pth is byte-identical to biubug6's mobilenet0.25_Final.pth. " | |
| "Outputs (per face): face_detection [x1,y1,x2,y2,score,(5 landmarks)], landmarks (98,2) or (68,2) px, emotion logits (8: Neutral,Happy,Sad,Surprise,Fear,Disgust,Anger,Contempt), " | |
| "gaze (yaw, pitch) 2-vector, action_units (8 AU intensities, tf_efficientnet_b0 MTL backbone). CLI TSV columns: timestamp,image_path,face_id,face_detection,landmarks,emotion,gaze_yaw,gaze_pitch,action_units.")) | |
| E.append(pip("openface-test", "0.1.26", "770c888c38f14b34000ac702b100b1bbb580f7394b12bc8ea320a3ecb2208c8e", OF3_LIC, OF3_LIC_URL, | |
| f"README install method: `pip install -r requirements.txt && pip install openface-test && openface download`. Its pins (numpy==1.26.4, scipy==1.13, Pillow==9.4.0, timm==1.0.15, " | |
| f"huggingface_hub==0.21.0, opencv_python==4.11.0.86) would downgrade the hci env, so installed with `pip install --no-deps openface-test==0.1.26` (torch 2.9.0.dev20250809+cu128 untouched) " | |
| f"plus its three missing unpinned-compatible deps scikit-image / imageio / tensorboardX. Adaptations needed in hci (baselines_dms/of3_compat.py): scipy.integrate.simps alias " | |
| f"(removed in scipy>=1.14), cfg_mnet['pretrain']=False (avoid CWD-relative ./weights/ load), LandmarkDetector(device_ids=[0]) on CUDA, STAR Base.init_instance patched to skip " | |
| f"the hard-coded /work/jiewenh/... tensorboardX writer + log file, and data_definition='300W' injected for the 68-point checkpoint (constructor hard-wires WFLW/98). Repo: https://github.com/CMU-MultiComp-Lab/OpenFace-3.0 @ {OF3_COMMIT}.", pip_args=["--no-deps"])) | |
| E.append(pip("scikit-image", "0.26.0", "7df650e79031634ac90b11e64a9eedaf5a5e06fcd09bcd03a34be01745744466", "BSD-3-Clause", "https://github.com/scikit-image/scikit-image/blob/main/LICENSE.txt", | |
| "installed 2026-09-14 for openface-test (skimage.transform); also pulled lazy-loader 0.5, tifffile 2026.9.9.")) | |
| E.append(pip("imageio", "2.37.4", "1ab2e22c8debf700f24c3ac43e8f95f3b3a8110c83b93411e97b4b0b2cd1c7e6", "BSD-2-Clause", "https://github.com/imageio/imageio/blob/master/LICENSE", "installed 2026-09-14 for openface-test.")) | |
| E.append(pip("tensorboardX", "2.6.5", "c10b891d00af306537cb8b58a039b2ba41571f0da06f433a41c4ca8d6abe1373", "MIT", "https://github.com/lanpa/tensorboardX/blob/master/LICENSE", "installed 2026-09-14 for openface-test (STAR conf imports SummaryWriter).")) | |
| # ---------------------------------------------------------------------------------------------- shared tooling (pre-existing, recorded for completeness) | |
| E.append(pip("gdown", "6.1.0", "38a36a94275b8272f684db469bbd73b4d1f64cbbc1751bcb993a1b2be8f013c8", "MIT", "https://github.com/wkentaro/gdown/blob/main/LICENSE", "pre-existing; used for the OpenFace 3.0 Drive folder (`--remaining-ok` flag does not exist in 6.1.0).")) | |
| E.append(pip("huggingface_hub", "1.17.0", "3b8156d23118e87f6a587648bfbc04f04a12a757ccb4ed298b35c4ae638bf24c", "Apache-2.0", "https://github.com/huggingface/huggingface_hub/blob/main/LICENSE", "pre-existing; hf_hub_download for VideoMAEv2 / L2CS mirrors.")) | |
| E.append(pip("timm", "1.0.27", "5ff07c9ddf53cbada88eab1c93ff175c64cab683b5a2fddf863bcee985926f89", "Apache-2.0", "https://github.com/huggingface/pytorch-image-models/blob/main/LICENSE", "pre-existing; timm.layers used by the vendored VideoMAEv2 model and by OpenFace MTL (tf_efficientnet_b0).")) | |
| E.append(pip("safetensors", "0.5.3", "cead1fa41fc54b1e61089fa57452e8834f798cb1dc7a09ba3524f1eb08e0317a", "Apache-2.0", "https://github.com/huggingface/safetensors/blob/main/LICENSE", "pre-existing; used only for the L2CS cross-check.")) | |
| E.append(pip("torch", "2.9.0.dev20250809+cu128", None, "BSD-3-Clause", "https://github.com/pytorch/pytorch/blob/main/LICENSE", | |
| "pre-existing nightly (not on PyPI -> no wheel sha256); MUST stay: verified unchanged after every install of this task.", url="https://download.pytorch.org/whl/nightly/cu128")) | |
| # ---------------------------------------------------------------------------------------------- public smoke-test sample (NOT BATON) | |
| E.append(dict(id="sample_portrait", kind="url", name="portrait.jpg", version="mediapipe-assets public sample (used by MediaPipe's own Face Landmarker examples)", | |
| url="https://storage.googleapis.com/mediapipe-assets/portrait.jpg", local_path="baselines_dms/weights/_samples/portrait.jpg", | |
| sha256="a6f11efaa834706db23f275b6115058fa87fc7f14362681e6abe14e82749de3e", size_bytes=176561, license="MediaPipe sample asset (Apache-2.0 repo assets)", | |
| license_url="https://github.com/google-ai-edge/mediapipe/blob/master/LICENSE", notes="1024x820 portrait for face-dependent smoke tests only (blendshape names, field names); no BATON frame is used by smoke.py.")) | |
| return E | |
| def main() -> int: | |
| reg = R.load() | |
| old = {e["id"]: e for e in reg.get("entries", [])} | |
| entries = build() | |
| bad = 0 | |
| for e in entries: | |
| for k in R.FIELDS: | |
| e.setdefault(k, None) | |
| e["installed_utc"] = (old.get(e["id"]) or {}).get("installed_utc") or R.utc_now() | |
| probs = R.verify(e) | |
| bad += bool(probs) | |
| print(R._fmt(e, probs)) | |
| R.save({"schema": 1, "generated_utc": None, "entries": entries}) | |
| print(f"wrote {R.REGISTRY_JSON} with {len(entries)} entries; {bad} failing verification") | |
| return int(bad > 0) | |
| if __name__ == "__main__": | |
| sys.exit(main()) | |
Xet Storage Details
- Size:
- 25.1 kB
- Xet hash:
- 61962147b97e6829347657dee2a05f304c3f6612700748cc83dc1c64194cdc0d
·
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.