美文网首页
Jetson TX2开发全纪录2-图像识别&摄像头目标识

Jetson TX2开发全纪录2-图像识别&摄像头目标识

作者: 南风无影 | 来源:发表于2018-04-12 18:04 被阅读975次

    工程 jetson-inference是基于Caffe模型;
    使用了TensorRT2.1做加速,识别单张图片是50ms,识别摄像头可以做到18FPS不卡, 我略作了些修改...


    1. 先把硬件马力全速启动

    a. 开启显卡小风扇.增加时钟限制

    sudo ~/jetson_clocks.sh
    

    b. 开启cpu为6核(默认是4核)

    sudo nvpmodel -m 0
    

    2. Cloning the Repo

    安装依赖库

    $ sudo apt-get install git cmake
    

    克隆代码:

    $ git clone https://github.com/307509256/jetson-inference
    

    编译:

    $ cd jetson-inference
    $ mkdir build
    $ cd build  
    $ cmake ../   #大概花半小时
    $ make 
    $ cd   
    

    图像识别,(imagenet)

    $ ./imagenet-console orange_0.jpg output_0.jpg
    

    如下,只得到分类和置信度:


    imagenet-orange.jpg
    $ ./imagenet-console granny_smith_1.jpg output_1.jpg
    
    imagenet-apple.jpg

    开启摄像头的目标识别(带坐标的)

    $ ./imagenet-camera googlenet           # to run using googlenet
    

    也可以把结果输出,过滤我们要的类:cellphone,把那行追加输出到result.txt:

    ./imagenet-camera googlenet | grep "cellphone" >> result.txt 
    
    Screenshot from 2018-04-12 17-53-35.png 2.png

    我们看到,googlenet识别的百分比都不高,可以换其他模型
    更多的模型下载: Download

    摄像头

    note: 默认使用 Jetson的 CSI camera. 如果用第三方的摄像头需要修改 DEFAULT_CAMER

    #define DEFAULT_CAMERA   -1 // -1 for onboard camera, or change to index of /dev/video V4L2 camera (>=0)
    

    Note: to reflect the /dev/video V4L2 device of your USB camera and recompile. The webcam model it's tested with is Logitech C920.

    相关文章

      网友评论

          本文标题:Jetson TX2开发全纪录2-图像识别&摄像头目标识

          本文链接:https://www.haomeiwen.com/subject/xyqfkftx.html