dd6fe347创建于 4月9日历史提交

文件	最后提交记录	最后更新时间
reid	init	4 年前
scripts	!4671 【fix】批量修改模型python版本，兼容环境上的python3.8版本 * fix python version	3 年前
.gitignore	init	4 年前
LICENSE	init	4 年前
PCB_amp.py	init	4 年前
README.md	fix link validity Co-authored-by: frozenleaves<914814442@qq.com> # message auto-generated for no-merge-commit merge: !7517 merge master into master fix link validity Created-by: frozenn Commit-by: frozenleaves Merged-by: ascend-robot Description: ## Motivation Please describe the motivation of this PR and the goal you want to achieve through this PR. ## Modification Please briefly describe what modification is made in this PR. ## Self-test (Optional) If modifications to this PR may cause/fix function/accuracy/performance DTSs/issues, a self-inspection record needs to be attached. ## BC-breaking (Optional) If there are compatibility issues, such as dependencies on cann/torch_npu versions, they need to be explained in the PR. ## Checklist Before PR: - [ ] The new code needs to comply with the Clean Code specification. - [ ] The PR content is self-checked, and the expression can be clear and the writing standardized After PR: - [ ] CLA has been signed and all committers have signed the CLA in this PR. - [ ] The ci-pipeline is passed, Code Check is passed. See merge request: Ascend/ModelZoo-PyTorch!7517	1 个月前
README_raw.md	!7376 optimize public_address_statement.md Merge pull request !7376 from 王凯宇/master	8 个月前
demo.py	init	4 年前
modelzoo_level.txt	init	4 年前
prof.py	init	4 年前
pthtar2onnx.py	init	4 年前
public_address_statement.md	!7376 optimize public_address_statement.md Merge pull request !7376 from 王凯宇/master	8 个月前
requirements.txt	[众智][PyTorch]整改模型中的requirements.txt文件，删除torch,apex Signed-off-by: bailang <bailang12@h-partners.com>	3 年前

PCB

This implements training of PCB on the Market-1501 dataset, mainly modified from syfafterzy/PCB_RPP_for_reID.

PCB Detail

As of the current date, Ascend-Pytorch is still inefficient for contiguous operations.Therefore, PCB is re-implemented using semantics such as custom OP.

Requirements

Install PyTorch (pytorch.org)

PyTorch版本：CANN 5.0.T205 PT>=20210618

pip install -r requirements.txt Note: pillow recommends installing a newer version. If the corresponding torchvision version cannot be installed directly, you can use the source code to install the corresponding version. The source code reference link: Suggestion the pillow is 9.1.0 and the torchvision is 0.6.0
Download the Market-1501 dataset from https://paperswithcode.com/dataset/market-1501
- ```
unzip Market-1501-v15.09.15.zip
```

Training

To train a model, run main.py with the desired model architecture and the path to the ImageNet dataset:

# training 1p accuracy
bash scripts/train_full_1p.sh --data_path=real_data_path

# training 1p performance
bash scripts/train_performance_1p.sh --data_path=real_data_path

# training 8p accuracy
bash scripts/train_full_8p.sh --data_path=real_data_path

# training 8p performance
bash scripts/train_performance_8p.sh --data_path=real_data_path

# Online inference demo
python demo.py --data_path real_data_path --device npu
## 备注： 识别前后图片保存到 `inference/` 文件夹下

# To ONNX
python pthtar2onnx.py

PCB training result

	mAP	AMP_Type	Epochs	FPS
1p-GPU	-	O2	1	568.431
1p-NPU	-	O2	1	571.723
8p-GPU	77.2	O2	60	3600.983
8p-NPU	77.5	O2	60	2750.401

公网地址说明

代码涉及公网地址参考 public_address_statement.md