YOLOv3调试运行过程

阳光雨露有代码 2024-03-07 01:33:42

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'

0 阅读:0