7ea8f2f9创建于 2024年11月28日历史提交

文件	最后提交记录	最后更新时间
change	init	4 年前
test	!2097 【自研】【BugFix】修改环境变量设置，统一使用内置环境变量 * 【自研】【BugFix】修改环境变量设置，统一使用内置环境变量 * Merge branch 'master' of https://gitee.com/liuzhuheng/ModelZoo-PyTorch * 重命名 ACL_PyTorch/contrib/audio/Waveglow 为 ACL_PyTorch/contrib/audio/Wav…	3 年前
Cascade_RCNN_R101.patch	init	4 年前
LICENSE	init	4 年前
README.md	!6851 [Clean Code] 修改芯片型号描述 Merge pull request !6851 from 施康/master	1 年前
get_info.py	init	4 年前
mmdetection_coco_postprocess.py	init	4 年前
mmdetection_coco_preprocess.py	init	4 年前
modelzoo_level.txt	init	4 年前
requirements.txt	init	4 年前

Casacde_RCNN_R101模型PyTorch离线推理指导

1 环境准备

安装必要的依赖，测试环境可能已经安装其中的一些不同版本的库了，故手动测试时不推荐使用该命令安装

pip3.7 install -r requirements.txt

获取，修改与安装开源模型代码

git clone https://github.com/open-mmlab/mmdetection.git   
cd mmdetection  
git reset a21eb25535f31634cef332b09fc27d28956fb24b --hard
pip3.7 install -v -e .
patch -p1 < ../Cascade_RCNN_R101.patch   
cd ..

利用提供的change文件夹中的patch文件，完成补丁操作，命令参考如下示例,请用户根据安装包位置自行修改：

cd change
patch -p0 /usr/local/python3.7.5/lib/python3.7/site-packages/mmcv/ops/deform_conv.py deform_conv.patch
cd ../

获取权重文件

从LINK下载cascade_rcnn模型权重文件
数据集
本模型使用coco2017的验证集验证
获取benchmark工具
将benchmark.x86_64或benchmark.aarch64放到当前目录

2 离线推理

执行时使npu-smi info查看设备状态，确保device空闲

bash test/pth2om.sh  
bash test/eval_acc_perf.sh --datasets_path=./