├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE.md ├── README.md ├── SECURITY.md ├── classification_pretrained ├── LICENSE.md ├── classmap.json ├── sample_image_paths.csv └── tao_image_classification.ipynb ├── classification_tf2 ├── byom_converters │ ├── LICENSE.md │ ├── byom_converter_mobilenetv2.ipynb │ ├── byom_converter_resnet18.ipynb │ └── mobilenetv2_utils │ │ ├── LICENSE.md │ │ ├── mobilenetv2.py │ │ └── mobilenetv2_0.5-eaa6f9ad.pth ├── byom_person │ ├── LICENSE.md │ ├── specs_mobilenetv2 │ │ ├── spec.yaml │ │ └── spec_retrain.yaml │ ├── specs_resnet18 │ │ ├── spec.yaml │ │ └── spec_retrain.yaml │ ├── stm32ai_tao_mobilenetv2.ipynb │ └── stm32ai_tao_resnet18.ipynb ├── onnx_utils.py └── tao_person │ ├── LICENSE.md │ ├── specs │ ├── spec.yaml │ └── spec_retrain.yaml │ └── stm32ai_tao_efficientnet_b0.ipynb ├── docs ├── TAO-STM32CubeAI.png ├── post_processing_node_yolov4_tiny.png ├── removed_nms_head.png └── sample_prediction.jpg └── yolo_v4_tiny ├── LICENSE.md ├── __init__.py ├── specs ├── yolo_v4_tiny_retrain_person.txt ├── yolo_v4_tiny_tfrecords_person_train.txt ├── yolo_v4_tiny_tfrecords_person_val.txt └── yolo_v4_tiny_train_person.txt ├── using_yolo_v4_tiny_with_stm32ai_modelzoo.ipynb ├── utils ├── generate_convert_coco_subset_to_kitti.py └── remove_nms.py └── yolo_v4_tiny.ipynb /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/SECURITY.md -------------------------------------------------------------------------------- /classification_pretrained/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_pretrained/LICENSE.md -------------------------------------------------------------------------------- /classification_pretrained/classmap.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_pretrained/classmap.json -------------------------------------------------------------------------------- /classification_pretrained/sample_image_paths.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_pretrained/sample_image_paths.csv -------------------------------------------------------------------------------- /classification_pretrained/tao_image_classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_pretrained/tao_image_classification.ipynb -------------------------------------------------------------------------------- /classification_tf2/byom_converters/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_converters/LICENSE.md -------------------------------------------------------------------------------- /classification_tf2/byom_converters/byom_converter_mobilenetv2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_converters/byom_converter_mobilenetv2.ipynb -------------------------------------------------------------------------------- /classification_tf2/byom_converters/byom_converter_resnet18.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_converters/byom_converter_resnet18.ipynb -------------------------------------------------------------------------------- /classification_tf2/byom_converters/mobilenetv2_utils/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_converters/mobilenetv2_utils/LICENSE.md -------------------------------------------------------------------------------- /classification_tf2/byom_converters/mobilenetv2_utils/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_converters/mobilenetv2_utils/mobilenetv2.py -------------------------------------------------------------------------------- /classification_tf2/byom_converters/mobilenetv2_utils/mobilenetv2_0.5-eaa6f9ad.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_converters/mobilenetv2_utils/mobilenetv2_0.5-eaa6f9ad.pth -------------------------------------------------------------------------------- /classification_tf2/byom_person/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_person/LICENSE.md -------------------------------------------------------------------------------- /classification_tf2/byom_person/specs_mobilenetv2/spec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_person/specs_mobilenetv2/spec.yaml -------------------------------------------------------------------------------- /classification_tf2/byom_person/specs_mobilenetv2/spec_retrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_person/specs_mobilenetv2/spec_retrain.yaml -------------------------------------------------------------------------------- /classification_tf2/byom_person/specs_resnet18/spec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_person/specs_resnet18/spec.yaml -------------------------------------------------------------------------------- /classification_tf2/byom_person/specs_resnet18/spec_retrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_person/specs_resnet18/spec_retrain.yaml -------------------------------------------------------------------------------- /classification_tf2/byom_person/stm32ai_tao_mobilenetv2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_person/stm32ai_tao_mobilenetv2.ipynb -------------------------------------------------------------------------------- /classification_tf2/byom_person/stm32ai_tao_resnet18.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/byom_person/stm32ai_tao_resnet18.ipynb -------------------------------------------------------------------------------- /classification_tf2/onnx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/onnx_utils.py -------------------------------------------------------------------------------- /classification_tf2/tao_person/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/tao_person/LICENSE.md -------------------------------------------------------------------------------- /classification_tf2/tao_person/specs/spec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/tao_person/specs/spec.yaml -------------------------------------------------------------------------------- /classification_tf2/tao_person/specs/spec_retrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/tao_person/specs/spec_retrain.yaml -------------------------------------------------------------------------------- /classification_tf2/tao_person/stm32ai_tao_efficientnet_b0.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/classification_tf2/tao_person/stm32ai_tao_efficientnet_b0.ipynb -------------------------------------------------------------------------------- /docs/TAO-STM32CubeAI.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/docs/TAO-STM32CubeAI.png -------------------------------------------------------------------------------- /docs/post_processing_node_yolov4_tiny.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/docs/post_processing_node_yolov4_tiny.png -------------------------------------------------------------------------------- /docs/removed_nms_head.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/docs/removed_nms_head.png -------------------------------------------------------------------------------- /docs/sample_prediction.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/docs/sample_prediction.jpg -------------------------------------------------------------------------------- /yolo_v4_tiny/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/LICENSE.md -------------------------------------------------------------------------------- /yolo_v4_tiny/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/__init__.py -------------------------------------------------------------------------------- /yolo_v4_tiny/specs/yolo_v4_tiny_retrain_person.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/specs/yolo_v4_tiny_retrain_person.txt -------------------------------------------------------------------------------- /yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_person_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_person_train.txt -------------------------------------------------------------------------------- /yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_person_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_person_val.txt -------------------------------------------------------------------------------- /yolo_v4_tiny/specs/yolo_v4_tiny_train_person.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/specs/yolo_v4_tiny_train_person.txt -------------------------------------------------------------------------------- /yolo_v4_tiny/using_yolo_v4_tiny_with_stm32ai_modelzoo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/using_yolo_v4_tiny_with_stm32ai_modelzoo.ipynb -------------------------------------------------------------------------------- /yolo_v4_tiny/utils/generate_convert_coco_subset_to_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/utils/generate_convert_coco_subset_to_kitti.py -------------------------------------------------------------------------------- /yolo_v4_tiny/utils/remove_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/utils/remove_nms.py -------------------------------------------------------------------------------- /yolo_v4_tiny/yolo_v4_tiny.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/STMicroelectronics/stm32ai-tao/HEAD/yolo_v4_tiny/yolo_v4_tiny.ipynb --------------------------------------------------------------------------------