[西南交通大学][高校贡献][PyTorch离线推理][AlignedReid]-初次提交-ModelZoo-PyTorch-AtomGit

已合并

[西南交通大学][高校贡献][PyTorch离线推理][AlignedReid]-初次提交 #1016

AtomGit-Bot创建于 2022年7月4日从refs/pull/1016/head 合入到 master

[西南交通大学][高校贡献][PyTorch离线推理][AlignedReid]-初次提交 #1016

已合并

AtomGit-Bot创建于 2022年7月4日

从refs/pull/1016/head合入到master

AtomGit-Bot

2022年7月4日

精度

	TOP1	TOP5
310精度	80.64	92.1
710精度	80.64	92.1

bs1和最优bs对精度无影响

性能

	310	310p	T4	310p(aoe)	310p/310(aoe)	310p/T4(aoe)
bs1	1441.70	1137.70	1016.17	1432.52	0.9936	1.4097
bs4	1930.08	2952.41	2238.83	3939.99	2.0413	1.7598
bs8	2160.46	3547.06	2517.50	4723.63	2.1864	1.8763
bs16	2150.62	3149.61	2737.99	4241.01	1.9719	1.5489
bs32	2208.22	2983.12	1529.43	3717.08	1.6832	2.4303
最优bs	2208.22	3547.06	2737.99	4723.63	2.1391	1.7252

bs1:经过aoe与aipp操作，调整其他参数后最优性能为1432.52
最优bs8：经过aoe之后310p大于310的1.2，310p大于T4的1.6，性能达标

自检报告
测试环境：310+310p+T4
数据预处理

python3.7 AlignedReID_preprocess.py ./market1501/images ./prep_bin

生成数据集info文件

python3.7 gen_dataset_info.py bin ./prep_bin ./alignedreid.info 128 256

设置环境变量

source /usr/local/Ascend/ascend-toolkit/set_env.sh

转动态onnx文件

python3.7 AlignedReID_pth2onnx.py ./Market1501_AlignedReID_300_rank1_8441.pth ./AlignedReID_bs.onnx

转om文件

atc --framework=5 --model=AlignedReID_bs.onnx --output=AlignedReID_bs${batchsize} --input_format=NCHW --input_shape="image:%{batchsize},3,256,128" --log=debug --soc_version=Ascend${chip_name} --out_nodes="Gemm_133:0;Reshape_127:0;Transpose_132:0"

推理

chmod u+x benchmark.x86_64

./benchmark.x86_64 -model_type=vision -device_id=0 -batch_size=1 -om_path=./AlignedReID_bs1.om -input_text_path=./alignedreid.info -input_width=128 -input_height=256 -output_binary=False -useDvpp=False

精度验证

python3.7 AlignedReID_acc_eval.py ./result/dumpOutput_device0

#第1次验收测试

Pull Request已成功合入, 合并人@

（感谢 AtomGit-Bot 的贡献）

A AtomGit-Bot

2022年7月4日指派了王姜奔参与评审

A AtomGit-Bot

2022年7月4日指派了 hanjun 参与评审

A AtomGit-Bot

2022年7月4日指派了魏利霞参与评审

A AtomGit-Bot

2022年7月4日添加了   ascend-cla/no 标签

A AtomGit-Bot

2022年7月4日指派了陈卫华参与评审

此处折叠了45条消息查看更多

AtlasAccount成员

2022年7月27日评论：

Build Result	✅ SUCCESS	#2739
DownLoad Code	✅ SUCCESS	>>>
Compile	✅ SUCCESS	>>>

liuzhuheng

2022年8月18日评论：

/lgtm

陈卫华

2022年8月19日合并了Pull Request（通过 Web 合并）

AtlasAccount成员

2022年8月19日评论：

🔵 何嘉峻 PR ModelZoo-PyTorch pipeline is running. Please wait a moment... (Link)

AtlasAccount成员

2022年8月19日评论：

Build Result	✅ SUCCESS	#3987
DownLoad Code	✅ SUCCESS	>>>
Compile	✅ SUCCESS	>>>