YOLOv3是一种目标检测的深度学习模型,此代码使用的深度学习框架是tensorflow,在调试运行过程成遇到了很多错误,不断调试解决后,终于运行成功了,在此记录一下运行的过程,以帮助朋友们节约时间、少走弯路。
运行环境:win10、python3.7
首先下载YOLOv3的预训练模型yolov3.weights:
https://pjreddie.com/media/files/yolov3.weights然后把下载的此文件放到源代码文件夹里面
然后打开CMD窗口,切换路径为源代码文件夹的路径,运行模型转换的程序:
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5运行成功的截图:
运行过程中遇到的问题:
2023-04-28 06:56:15.382912: I tensorflow/stream_executor/cuda/cuda_driver.cc:831] failed to allocate 4.35G (4674388480 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
解决方法:
在运行文件里面加上代码:
import os
os.environ['CUDA_VISIBLE_DEVICES']='2, 3'