Yolo_踩的坑

Yolo_踩的坑

Config:

数据集处理:

  • OS:Mac OS X 10.14

跑模型:

  • OS:Ubuntu 16.04
  • GPU:GetForce GTX 960M
  • NVIDIA Driver:430.09
  • Cuda:10.1
  • Cudnn:7.5.1

安装并没有太多问题,准备跑自己的模型的时候还是遇到了蛮多问题

推一下官方教程:https://pjreddie.com/darknet/yolo/

1. cuda error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0’ failed.

改一下cfg,减小batch,增大subdivisions

给咱来个🍰,啾咪