ESP-Detection is a lightweight and ESP-optimized project based on Ultralytics YOLOv11, designed for real-time object detection on ESP series chips. It allows you effortlessly train a detection model for specific target and deploy the model on chips easily by ESP_DL.
ESP-Detection provides a series of ultra-lightweight models along with APIs that enables you to train custom detection models tailored to your specific use cases. The offered models are optimized for efficient deployment on ESP AI chips, like ESP32-P4 and ESP32-S3. The project also includes example applications such as cat detection, dog detection, and pedestrian detection.
rect=True
for training and deployment on non-square image resolutions, improving both accuracy and speed.conda create -n espdet python=3.8 conda activate espdet pip install -r requirements.txt
release/v5.3
or above.Dataset format in esp-detection follows the YOLO detection dataset format. To convert your existing dataset from other formats (like COCO etc.) to YOLO format, please use converter.py by Ultralytics.
Step 2: Train and deploy the modelIn esp-detection, we provide an all-in-one script espdet_run.py
that streamlines the entire flow. With a single command, users can easily perform model training, export, quantization, and deployment. You can customize the espdet_run.sh
shell script and execute the full pipeline with sh espdet_run.sh
.
✨ We have integrated rect=True
training and deployment into espdet_run.py
. You can simply set rect=True
and specify size=[h, w]
to enable it. For more details, please refer to the tutorial.
python espdet_run.py \ --class_name mycat \ --pretrained_path None \ --dataset "cfg/datasets/coco_cat.yaml" \ --size 224 224 \ --target "esp32p4" \ --calib_data "deploy/cat_calib" \ --espdl "espdet_pico_224_224_mycat.espdl" \ --img "espdet.jpg"
Once Step 2 is completed, an example will be automatically generated. You can use it to perform model inference on ESP32-S3 or ESP32-P4. Similar to other esp-dl/examples, it enables customized model inference on a single input image and prints the results to the terminal.
You can run the example directly in the terminal with the following command:
cd path/to/custom_example idf.py set-target esp32p4 idf.py flash monitor
Please submit an issue if you find any problems using our products, and we will reply as soon as possible.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4