2.3 KiB
2.3 KiB
Übergabe Hailo
- Raphael Maehnle
- Thomas Hamböck
Agenda
- CVAT: Login, Data Import, Labeling, Data export
- Create a new Project; Important: Labels eintragen, möglichst in konsistenter Reihenfolge (zu anderen Projekten)
- Früher gab es nur Tasks => da war Labels konsistent halten aufwändiger; Anm.: Projects funktionieren zZ noch nicht stabil
- Anm.: Früher gab es ein Speicherproblem; Bilder blieben im RAM => häufigeres Seite neu laden sinnvoll
- Jobs: Aufteilung von Bildern auf annotierende Leute (User Assignment)
- Bilder annotieren: Tastenkürzel ("n" > neue Bounding Box, ...): https://openvinotoolkit.github.io/cvat/docs/manual/advanced/shortcuts/
- "..." > export as dataset > Yolo1.1
- Create a new Project; Important: Labels eintragen, möglichst in konsistenter Reihenfolge (zu anderen Projekten)
- hailo setup for development (training)
- prerequisite: nvidia-docker
- clone hailo model zoo: https://github.com/hailo-ai/hailo_model_zoo
- download & run docker container "Hailo Software Suite (for Ubuntu 20.04)"
- note: volume mount is missing in the setup guide => will be required for importing data
- Retraining Yolov5m
- Tutorial for yolov5 training https://github.com/hailo-ai/hailo_model_zoo/blob/master/docs/RETRAIN_ON_CUSTOM_DATASET.md
- dataset meta setup: http://192.168.0.63:8930/raphael.maenle/hailo_inference/-/blob/main/dataset/dataset/dataset.yaml labels aus CVAT nochmal in die liste übertragen, nc == len(names)
- from readme:
docker run -it --gpus all -ipc=host -v /path/to/dataset/:/dataset yolov5:v0
based on https://github.com/hailo-ai/hailo_model_zoo/blob/master/training/yolov5/Dockerfile - https://github.com/hailo-ai/hailo_model_zoo/blob/master/docs/RETRAIN_ON_CUSTOM_DATASET.md?plain=1#L183
epochs 3 ist etwas wenig; ~100 wäre besser
--data
=> auf dataset - Export to ONNX: https://github.com/hailo-ai/hailo_model_zoo/blob/master/docs/RETRAIN_ON_CUSTOM_DATASET.md?plain=1#L192 For visualisation: ONNX viewer, e.g.: https://github.com/lutzroeder/Netron
- Yolov5 hailo quantisation + compile
hailo_sw_suite_docker_run.sh
=> add-v
asDOCKER_ARGS
for volume mount in script- network config:
yolov5m.yaml
- do
compile
=> includesquantize
.hef
=> network bitstream
- hailo setup for deployment (inference)
HailoRT
=> setup für runtime
- Inference