J'ai essayé la détection d'objets avec YOLO v3 (TensorFlow 2.1) sur le GPU de windows!

environnement

windows7 64bit Gefore GTX 680MX GPU anaconda

1. 1. Installation de CUDA

Téléchargez la version de mise à jour 2 de CUDA Toolkit 10.1 à partir du site suivant (TensorFlow 2.1.0 est pris en charge à cause de CUDA Toolkit 10.1) https://developer.nvidia.com/cuda-toolkit-archive

Ne sélectionnez pas l'intégration de Visual Studio pour l'installation

Si l'installation réussit, exécutez nvcc -V et le résultat suivant sera affiché.

C:\Windows\System32>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.1, V10.1.243

2. Installation de cuDNN

Sélectionnez et téléchargez cuDNN pour CUDA 10.1 à partir du site suivant (Nécessite un compte NVIDIA) https://developer.nvidia.com/cudnn

Décompressez le fichier zip et Renommez le dossier cuda en cuda765 et Copiez dans le chemin "C: \ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v10.1"

Après la copie 1.JPG

3. Paramètres des variables d'environnement

Ajoutez le chemin cuDNN suivant à PATH

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\cuda765\bin

4. Créez keras_work avec conda

conda create -n keras_work
activate keras_work

5. Installez tensoflow

conda install tensorflow

6. Confirmation

(base) C:\Users\mac>activate keras_work

(keras_work) C:\Users\mac>python
Python 3.7.7 (default, Apr 15 2020, 05:09:04) [MSC v.1916 64 bit (AMD64)] :: Ana
conda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
2020-05-06 19:46:04.480481: I tensorflow/stream_executor/platform/default/dso_lo
ader.cc:44] Successfully opened dynamic library cudart64_101.dll
>>> print(tf.__version__)
2.1.0
>>> print(tf.test.is_gpu_available())
・ ・ ・
physical GPU (device: 0, name: GeForce GTX 680MX, pci bus id: 0000:01:00.0, compute capability: 3.0)
True

7. Préparation du matériau YOLO V3

Obtenir la source de Github

cd c:\temp
git clone https://github.com/zzh8829/yolov3-tf2.git
cd yolov3-tf2

Téléchargez yolo3.weight sur pjreddie.com

wget https://pjreddie.com/media/files/yolov3.weights --no-check-certificate

Comme il est lent d'obtenir des poids yolov3.weights à partir de pjreddie.com ci-dessus, vous pouvez également télécharger l'URL suivante https://pan.baidu.com/s/1G2Qh-V8kyLOq4oDbTwK6HQ Proposition (mot de passe): vogw Le fichier est "yolo_tf2.1 \ data \ yolov3.weights"

Déplacez le fichier yolo3.weight vers le chemin yolov3-tf2 Confirmez que le fichier a été déplacé

(keras_work) C:\temp\yolov3-tf2>dir /B *.weights
yolov3.weights

8. Convertir (convertir le poids darknet pré-entraîné)

python convert.py --weights ./yolov3.weights --output ./checkpoints/yolov3.tf

9. détection

python detect.py --image ./data/street.jpg

output.jpg

Recommended Posts

J'ai essayé la détection d'objets avec YOLO v3 (TensorFlow 2.1) sur le GPU de windows!
J'ai essayé la détection d'objets avec YOLO v3 (TensorFlow 2.0) sur un processeur Windows!
J'ai essayé de trouver la moyenne de plusieurs colonnes avec TensorFlow
[Pour les débutants] J'ai essayé d'utiliser l'API Tensorflow Object Detection
J'ai essayé d'exécuter YOLO v3 avec Google Colab
J'ai essayé de jouer avec la calculatrice avec tkinter
J'ai couru le tutoriel TensorFlow avec des commentaires (classification du texte des critiques de films)
J'ai essayé la nouvelle API de détection d'objets de tensorflow sur macOS Sierra
J'ai essayé de trouver l'entropie de l'image avec python
J'ai essayé la "correction gamma" de l'image avec Python + OpenCV
2020/02 Python 3.7 + TensorFlow 2.1 + Keras 2.3.1 + YOLOv3 Détection d'objets avec la dernière version
J'ai essayé de refactoriser le modèle CNN de TensorFlow en utilisant TF-Slim
J'ai essayé d'exécuter PIFuHD sur Windows pour le moment
J'ai essayé de remplacer le disque dur Windows 10 par un SSD plus petit
Jusqu'au début du tutoriel django avec pycharm sous Windows
J'ai essayé de changer le script python de 2.7.11 à 3.6.0 sur Windows10
Je suis tombé sur TensorFlow (Quelle est la mémoire du GPU)
[Pour ceux qui veulent utiliser TPU] J'ai essayé d'utiliser l'API de détection d'objets Tensorflow 2
J'ai essayé de trier les objets de l'image du plat de steak-① Détection d'objets
J'ai essayé un réseau de neurones convolutifs (CNN) avec un tutoriel TensorFlow sur Cloud9-Classification des images manuscrites-
Détection de visage avec YOLO Face (Windows10, Python3.6)
J'ai essayé le tutoriel TensorFlow 1er
J'ai essayé le tutoriel TensorFlow 2ème
J'ai essayé la détection de visage avec MTCNN
Utilisez Tensorflow 2.1.0 avec Anaconda sur Windows 10!
J'ai essayé de gratter le classement du calendrier de l'avent Qiita avec Python
J'ai essayé d'automatiser l'arrosage du pot avec Raspberry Pi
J'ai essayé la validation croisée basée sur le résultat de la recherche de grille avec scikit-learn
J'ai essayé d'agrandir la taille du volume logique avec LVM
J'ai essayé d'exécuter la partie DNN d'OpenPose avec le processeur Chainer
J'ai vérifié l'image de l'Université des sciences sur Twitter avec Word2Vec.
J'ai essayé d'améliorer l'efficacité du travail quotidien avec Python
J'ai essayé d'exécuter le didacticiel TensorFlow avec des commentaires (_TensorFlow_2_0_Introduction pour les débutants)
J'ai essayé de créer Othello AI avec tensorflow sans comprendre la théorie de l'apprentissage automatique ~ Introduction ~
J'ai essayé de créer Othello AI avec tensorflow sans comprendre la théorie de l'apprentissage automatique ~ Implémentation ~
Tutoriel TensorFlow J'ai essayé MNIST 3rd
J'ai essayé de transformer l'image du visage en utilisant sparse_image_warp de TensorFlow Addons
J'ai essayé le serveur asynchrone de Django 3.0
J'ai essayé d'implémenter Autoencoder avec TensorFlow
J'ai essayé tensorflow pour la première fois
J'étais accro à l'exécution de tensorflow sur GPU avec le pilote NVIDIA 440 + CUDA 10.2
J'ai essayé de visualiser AutoEncoder avec TensorFlow
J'ai essayé d'installer TensorFlow (version GPU) sur Ubuntu
J'ai essayé d'obtenir le code d'authentification de l'API Qiita avec Python.
J'ai essayé de résumer la manière logique de penser l'orientation objet.
J'ai essayé avec les 100 meilleurs packages PyPI> J'ai essayé de représenter graphiquement les packages installés sur Python
J'ai essayé de jouer avec l'image avec Pillow
J'ai essayé d'analyser la négativité de Nono Morikubo. [Comparer avec Posipa]
J'ai essayé de rationaliser le rôle standard des nouveaux employés avec Python
J'ai essayé de visualiser le texte du roman "Weather Child" avec Word Cloud
J'ai construit un environnement TensorFlow avec windows10
J'ai essayé d'obtenir les informations sur le film de l'API TMDb avec Python
Détection d'objets de cuisson par classification d'images Yolo +
J'ai essayé de prédire le comportement du nouveau virus corona avec le modèle SEIR.
J'ai essayé la détection 3D d'une voiture
J'ai essayé de créer Othello AI avec tensorflow sans comprendre la théorie de l'apprentissage automatique ~ Battle Edition ~
J'ai installé Pygame avec Python 3.5.1 dans l'environnement de pyenv sur OS X
J'ai essayé le tutoriel TensorFlow (MNIST pour les débutants) sur Cloud9-Classification des images manuscrites-
J'ai essayé de visualiser facilement les tweets de JAWS DAYS 2017 avec Python + ELK
J'ai essayé d'exécuter le didacticiel de détection d'objets en utilisant le dernier algorithme d'apprentissage en profondeur