Skip to content

zt15905696693/yolov3_opencl

 
 

yolov3 的 opencl 实现

Darknet Logo

Darknet

Darknet is an open source neural network framework written in C and CUDA. It is fast, easy to install, and supports CPU and GPU computation.

For more information see the Darknet project website.

For questions or issues please use the Google Group.

usage

使用opencl进行计算,在Makefile里设置OPENCL=1,执行make即可

OpenCL代码主要在src/opencl_tool.h, src/opencl_tool.c里,实现OpenCL平台上下文和设备选择,内存分配及主机到设备和设备到主机的迁移。

src/opencl_kernel.cl,包含所有设备端__kernel函数的实现。

wget https://pjreddie.com/media/files/yolov3.weights

./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

./darknet detector train cfg/coco.data cfg/yolov3.cfg darknet53.conv.74

./darknet detector valid cfg/coco.data cfg/yolov3.cfg yolov3.weights

Releases

No releases published

Packages

No packages published

Languages

  • C 91.9%
  • Cuda 6.3%
  • Other 1.8%