Série yolov5 (i) - - netron, un outil de visualisation de réseau

Un.、Introduction

netron：Outils de visualisation web,L'utilisation de ce logiciel peut clairement montrer l'entrée et la sortie de chaque couche du modèle de réseau,L'architecture globale du réseau,Et prend en charge différents cadres de réseau.
Comme ci - dessous：

2.、Étapes d'installation（IntroductionwindowsInstallation du système）

（D'autres versions sont disponibles en référence àBlogs）

1、Ouvrez ce lien：GithubTélécharger le lien

Ouvrir le lien ci - dessus,En bas. install,Sélectionner windows Télécharger

2、Sélectionnez l'image suivanteNetron-Setup-5.8.9.exe,Télécharger l'installation

3、Une fois installé avec succès, il apparaît comme suit：

Trois、Ouvre.yolov5Modèle de série

1、Conversion

Parce quenetronC'est exact. .pt Le fichier pour n'est pas très compatible , Il faut donc convertir le fichier ,Inyolov5 En ingénierie,Offreexport.py, Conversion de fichiers possible

2、 Saisissez la commande suivante directement au terminal

python export.py --weights yolov5s.pt --include onnx

Attention!： Attention lors de l'entrée des commandes export.py Et yolo5s.pt Chemin vers
Je suis directement dans le Répertoire du fichier , Il n'est donc pas nécessaire d'écrire d'autres chemins

Après avoir exécuté la commande,Il sera là.yolov5s.pt Générer sous le répertoire des pairs yolov5s.onnx
Les outils de visualisation Web sont maintenant disponibles netronOuvre.

Quatre、 Problèmes possibles avec l'exécution des commandes

requirements: onnx not found and is required by YOLOv5, attempting auto-update...
requirements: 'pip install onnx' skipped (offline)
ONNX: export failure: No module named 'onnx'

Solutions

Installationonnx
pip install onnx

Il n'y a aucun problème à exécuter les commandes ci - dessus après