Ii-robot!7376 optimize public_address_statement.md

9549f127创建于 2025年9月4日历史提交

文件	最后提交记录	最后更新时间
data	init	4 年前
layers	!4685 [fix] 修改算子调用方式 * fix code clean all_py about Operator modification * fix code clean * fix op way first	2 年前
models	init	4 年前
test	!4671 【fix】批量修改模型python版本，兼容环境上的python3.8版本 * fix python version	3 年前
utils	init	4 年前
README.md	!5200 RefineDet、RetinaMask、RetinaNet 模型公网地址整改 * 局部变量名小写 * RefineDet、RetinaMask、RetinaNet 模型公网地址整改	2 年前
demo.py	init	4 年前
eval_refinedet.py	init	4 年前
modelzoo_level.txt	init	4 年前
public_address_statement.md	!7376 optimize public_address_statement.md Merge pull request !7376 from 王凯宇/master	8 个月前
requirements.txt	!848 [众智][PyTorch]RefineDet模型训练脚本整改 * update PyTorch/contrib/cv/detection/RefineDet/test/train_full_1p.sh. * [众智][PyTorch]RefineDet模型训练脚本整改	3 年前
train_1p.py	!5200 RefineDet、RetinaMask、RetinaNet 模型公网地址整改 * 局部变量名小写 * RefineDet、RetinaMask、RetinaNet 模型公网地址整改	2 年前
train_8p.py	!5200 RefineDet、RetinaMask、RetinaNet 模型公网地址整改 * 局部变量名小写 * RefineDet、RetinaMask、RetinaNet 模型公网地址整改	2 年前
url.ini	!5200 RefineDet、RetinaMask、RetinaNet 模型公网地址整改 * 局部变量名小写 * RefineDet、RetinaMask、RetinaNet 模型公网地址整改	2 年前

#RefineDet模型PyTorch 训练指导

1 环境准备

1.安装必要的依赖

pip3 install -r requirements.txt 注：pillow建议安装较新版本，与之对应的torchvision版本如果无法直接安装，可使用源码安装对应的版本，源码参考链接：https://github.com/pytorch/vision 建议：Pillow版本是9.1.0 torchvision版本是0.6.0

2.获取数据集

sh data/scripts/VOC2007.sh
sh data/scripts/VOC2012.sh

下载好的数据集位于 ./data/VOCdevkit

2 训练

路径要写到 VOCdevkit

# npu env
source test/env_npu.sh

# 1p train perf
bash test/train_performance_1p.sh --data_path=xxx

# 路径要写到 VOCdevkit
# 例如

bash test/train_performance_1p.sh --data_path=./data/VOCdevkit

# 8p train perf
bash test/train_performance_8p.sh --data_path=xxx

# 8p train full
bash test/train_full_8p.sh --data_path=xxx 

# 8p eval
bash test/train_eval_8p.sh --data_path=xxx 

# finetuning
bash test/train_finetune_1p.sh --data_path=xxx 

# online inference demo 
python3 demo.py

公网地址说明

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