目标检测YOLOv3笔记(2):尝试训练数据
接下来是训练自己的模型
自己标注数据
Ubuntu下载LabelImg标注工具
git clone https://github.com/tzutalin/labelImg
安装依赖
apt-get install pyqt4-dev-tools
apt-get install python-lxml
编译make all
,运行
python labelImg.py
接下来是训练自己的模型
Ubuntu下载LabelImg标注工具
git clone https://github.com/tzutalin/labelImg
安装依赖
apt-get install pyqt4-dev-tools
apt-get install python-lxml
编译make all
,运行
python labelImg.py
网络原理就不多讲,我也是半知半解,我们先关注怎么用。
下载源码
git clone https://github.com/pjreddie/darknet.git
对Makefile的几个参数根据自己的情况进行修改
GPU=1
CUDNN=1
OPENCV=1
编译make
,下载训练好的权重文件:wget https://pjreddie.com/media/files/yolov3.weights
然后就可以测试
测试图片
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
测试视频,video file
自己放段视频
./darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights <video file>