search

espressif / esp-nn

Optimised Neural Network functions for Espressif chipsets
Apache License 2.0
125 stars 22 forks source link

Ocr model that trained by yolov5 #12

Open WrinkLeeTR opened 2 months ago

WrinkLeeTR commented 2 months ago

Hi Everyone,

I have same models that trained with yolov5 but some parameters was changed. My question is can I include OCR model that was trained with yolov5?

vikramdattu commented 2 months ago

@WrinkLeeTR Currently, esp-nn can work only with tflite-micro model format. Convert your model to the tflite-micro format and fit it into a format: https://github.com/espressif/esp-tflite-micro esp-nn optimisations would then be automatically used there.

WrinkLeeTR commented 2 months ago

@vikramdattu thanks for your reply. I used tflite micro and I can include my model. But I got error with esp32s3 when the devices start to run. I always got memory error. The tflite file has 1mb size. How can I solve that problem

vikramdattu commented 2 months ago

In the main_functions.cc° file, you can increase TensorArena size and allocate the buffer on SPIRAM. s3-eye has 4MB of SPIRAM.