工业级别的目标检测关注的不仅仅是精度,还有速度,能达到实时是最理想状态,一般来讲,目标检测实时大于12.5fps被认为是实时,针对TX2利用yolov4检测博主做了一个详细的调研和测试。
1.下载darknet,网址如下:
git clone https://github.com/AlexeyAB/darknet.git
2.配置makefile文件
由于TX2已经刷机Jetpack4.4,TX2里面有gpu,cuda和cudnn等,修改makefile文件如下:
GPU=1
CUDNN=1
OPENCV=1
3.在darknet路径下编译如下
make
4.下载权重,放到darknet目录下
# yolov4-tiny.weights
wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights
# yolov4.weights
wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4.weights
5.测试(yolov4 and yolov4-tiny)
(1).测试图片
./darknet detector test cfg/co