2024-07-11
한어Русский языкEnglishFrançaisIndonesianSanskrit日本語DeutschPortuguêsΕλληνικάespañolItalianoSuomalainenLatina
Viitedokumentaatio: https://docs.espressif.com/projects/esp-dl/zh_CN/latest/esp32/tutorials/index.html
Tämä tapaus esittelee koko TVM-käyttöönottomallin käyttöprosessin. Tämä projekti perustuu TVM v0.14.0 -haaraan ja on kokeellisessa tilassa. Tällä hetkellä ei ole suunnitteilla jatkuvaa ylläpitoa. Tällä hetkellä vain ESP-DL:n conv2d-operaattori on kytkettynä, ja muut operaattorit voivat aiheuttaa poikkeuksia.
Valmistella
ESP-DL on syvän oppimisen päättelykehys, joka on mukautettu ESP-sarjan siruihin. Tämä kirjasto ei voi suorittaa mallikoulutusta. Käyttäjät voivat kouluttaa mallia TensorFlow-, PyTorchi- ja muiden koulutusalustojen avulla ja ottaa mallin käyttöön ESP-DL:n kautta.
Erityinen prosessi on
Muunna ensin opetettu malli, kuten tensorflowPyTorch. Ottaen esimerkkinä TensorFlow-alustan, voit käyttää skriptissä tf2onnx-komentoa muuntaaksesi opetetun TensorFlow-mallin ONNX-mallimuotoon.
Sitten float32-mallille suoritetaan sarja operaatioita sen valmistelemiseksi kvantifiointia varten.
Käytä sitten kvantisointityökalua esikäsitellyn float32-mallin hyväksymiseen syötteeksi ja luo int8-kvantisointimalli.
Vaihe 2.1: Valmistaudu syöttämään
Valmistele syöttökuva Syötekuvan koon tulee olla yhdenmukainen saadun ONNX-mallin syöttökoon kanssa. Mallin syöttökokoa voi tarkastella Netron-työkalun kautta.
Vaihe 2.2: Käyttöönottoprojektin luominen
Käytä TVM:ää luodaksesi automaattisesti projektin mallipäätelmän suorittamista varten.
Vaihe 3.1: Suorita päättely
Edellisessä vaiheessa luodun projektitiedoston new_project rakenne on seuraava:
├── CMakeLists.txt
├── komponentit
│ ├── esp-dl
│ └── tvm_model
│ ├── CMakeLists.txt
│ ├── crt_config
│ └── malli
├── pääasiallinen
│ ├── app_main.c
│ ├── input_data.h
│ ├── output_data.h
│ └── CMakeLists.txt
├── partitions.csv
├── sdkconfig.defaults
├── sdkconfig.defaults.esp32
├── sdkconfig.defaults.esp32s2
├── sdkconfig.defaults.esp32s3
Kun olet määrittänyt terminaalin ESP-IDF-ympäristön (huomaa ESP-IDF:n versio), voit suorittaa projektin:
idf.py set-target esp32s3
idf.py flash-näyttö
Vaihe 3.2: Virheenkorjaus