与参考文章大部分都是相似的,如果不习惯看英文,可以看看我下面的描述
上篇博客:在Jetson TX2上捕获、显示摄像头视频 介绍了一个脚本 tegra-cam.py。主要是在Jetson TX2上显示三种摄像头捕获的视频。本节将对脚本进行扩展,使用caffe分类模型对摄像头视频进行演示。
主要的测试环境:python3和Jetson TX2
先决条件
caffe模型采用默认的 bvlc_reference_caffenet 需要下载预训练模型和标签。下载连接 将下载好的模型拷到 /home/nvidia/caffe/models/bvlc_reference_caffenet/ 文件夹。执行
cd /home/nvidia/caffe
./data/ilsvrc12/get_ilsvrc_aux.sh
代码的下载和使用
使用USB摄像头,分辨率设为1280x720。'–vid 1' 意味着 /de/video1
python3 tegra-cam-caffe.py --usb --vid 1 --width 1280 --height 720
使用板载摄像头
python3 tegra-cam-caffe.py
使用IP CAM,同时替换成你自己的IP
python3 tegra-cam-caffe.py --rtsp --uri rtsp://admin:XXXXXX@192.168.1.64:554
使用板载摄像头对不同caffe模型进行图像分类
python3 tegra-cam-caffe.py --prototxt XXX.prototxt \--model YYY.caffemodel \--labels ZZZ.txt \--mean UUU.binaryproto