Views
No views yet
An int8-quantized YOLOv8n fine-tuned to detect 24 classes of sidewalk obstacles (bollards, tree trunks, movable signage, poles, and more) that are absent from COCO. 3.5 MB, built to run on-device.
| 기반 모델 | YOLOv8n (Ultralytics 8.4.101) — COCO 사전 학습 가중치에서 추가 학습 |
| 형식 | ONNX opset 17, int8 양자화 전체 가중치 |
| 용량 | 3.5 MB (float32 원본 12.3 MB) |
| 입력 | images — float32 [1, 3, 640, 640], RGB, 0~1 정규화, 레터박스 |
| 출력 | output0 — float32 [1, 28, 8400]. 28 = 박스 4(cx, cy, w, h) + 클래스 점수 24 |
| 클래스 | 24종 |
| 라이선스 | AGPL-3.0-only |
| 학습과 변환 코드 | https://github.com/J4VIS-2026/nrh-train |
[!IMPORTANT] 이model.onnx는 int8 양자화본입니다. ONNX 관례에서model.onnx는 보통 float32를 뜻하지만, 이 저장소는 단말 배포용 int8만 공개합니다. 입출력 텐서 자체는 float32이고 양자화는 내부 가중치에 적용되어 있어, 쓰는 쪽에서 따로 처리할 것은 없습니다.NMS가 들어 있지 않습니다. 후처리를 직접 하셔야 합니다.
person car bus truck motorcycle micromobility push_cart wheelchair petbollard pole tree_trunk barricade bench chair table potted plant
fire hydrant movable_signage traffic_sign traffic light control_cabinet
service_terminal transit_stopmicromobility는 자전거와 킥보드를 합친 것이고, pet은 개와 고양이를 합친 것입니다.names에 들어 있습니다. 코드에 직접 나열하지
말고 거기서 읽으시면 모델을 바꿔도 깨지지 않습니다.114/255로 채웁니다 (레터박스)[1, 3, 640, 640] float32cx, cy, w, h 형식이므로 모서리 좌표로 바꾸고, 레터박스를 되돌려 원본 크기로
복원합니다conf 0.001 기준입니다.| mAP50 | mAP50-95 | |
|---|---|---|
| float32 | 0.6928 | 0.4742 |
| int8 (여기서 배포하는 것) | 0.6832 | 0.4539 |
| 양자화 손실 | -1.4% | -4.3% |
pole과 tree_trunk처럼 인접한 물체가 한
박스에 묶였다가 풀리는 현상이 관찰됩니다.pole↔tree_trunk, car↔truck처럼 형태가 비슷한 쌍에서
프레임마다 판정이 바뀔 수 있습니다.이 결과물은 AI 허브의 「인도보행 영상」 데이터셋을 활용하였습니다.
LICENSE에 있습니다.