本项目在搭载Sophon BM168X TPU的开发板的linux环境中实现了DETR目标检测模型的部署,并可用Gradio Web Demo的形式进行使用。
所需文件 | 百度网盘 |
---|---|
模型权重 | 链接 密码:zzy7 |
sophon-sail依赖包 | 链接 密码:jq05 |
请在搭载Sophon BM168X TPU的soc的linux环境中进行如下步骤:
下载sophon-sail的python依赖包,并进行安装。
pip3 install path/to/sophon_arm-3.6.0-py3-none-any.whl
克隆本项目到soc的linux环境的相应目录下。
git clone https://github.com/MINGtoMING/DETR-Web-Demo.git
安装相关依赖:
pip install -r requirements.txt
下载转化好的DETR的模型权重压缩包,并解压到本项目的目录下:
tar -zxvf /path/to/checkpoints.tar.gz
查看本设备的ip地址, 假设为192.168.11.3
:
ifconfig
运行gradio demo:
python3 app.py --server-name="0.0.0.0" --server-port=8888
最后在与本设备在同一局域网下的主机浏览器中输入192.168.11.3:8888
进行访问即可。