NoseNet / weights /README.md
LiFLabUC's picture
Add NoseNet model cards and checkpoints
1dbc4c6 verified
|
Raw History Blame Contribute Delete
383 Bytes
# Checkpoint layout
- `face_visibility/best.pt`: YOLOv5m face/no-face checkpoint. Class mapping: `0 = face`, `1 = no_face`.
- `nose_detection/best.pt`: YOLOv5m nose-localization checkpoint. Class mapping: `0 = nose`.
These are PyTorch checkpoint files. Load them only in a trusted Python environment and use the compatible YOLOv5 code distributed with the NoseNet implementation.