我的AI之路(46)--在Xavier上的自制dock镜像内安装支持部署EfficientDet和CenterNet模型以及mmdetection序列模型的环境

本文详述如何在ARM64版Docker环境中配置GPU支持,安装EfficientDet、CenterNet及mmdetection工具,包括软件包安装、源码编译与模型测试过程。

      按照制作支持GPU的用于部署AI模型的ARM64版docker镜像一文中的步骤制作出基础镜像并创建容器后,连到这个容器,执行下列步骤,进行支持部署EfficientDet和CenterNet模型以及mmdetection序列模型的环境的安装:

1) EfficientDet

    apt-get update
    sudo apt-get install apt-file 
    apt-file update
    apt-file search libgthread-2.0.so.0

    apt-get install libglib2.0-dev
    apt-get install libsm6
    apt-get install libxrender1
    apt-get install libxext6
    pip3 install -i  http://mirrors.aliyun.com/pypi/simple/  tensorboardX  --trusted-host  mirrors.aliyun.com

下载efficientdet源码并根据自己的数据集的分类情况对代码做出相应的修改,具体办法会在后续文章中给出,并将标注好了的验证测试数据在data/coco/val2017下面,将训练出的模型放到test_dataset.py/mAP_evaluation.py脚本里指定的地方,然后即可执行下面的测试以检查GPU是否使用正常,并可在test_dataset.py中增加时间的打印检查识别每张图片所花费的时间:

     cd efficientd

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Arnold-FY-Chen

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值