昇腾社区首页
中文
注册

SSD快速迁移学习

SSD快速迁移学习没有新增文件夹,在原有SSD基本检测功能代码中添加额外命令参数来开启该功能,推荐在“ssd_mobilenet_fpn_mindspore”文件夹中进行。

训练命令参考

训练启动参考以下命令。

python3 model_train.py --train_dataset_path={train_dataset_path} --train_output_path={train_output_path} --pretrained_ckpt_path={pretrained_ckpt_path} --epoch_size=100 --batch_size=4 --input_width=2048 --input_height=1536 --init_lr=0.001 --device_num=0 --run_eval=True --quick_train=True --eval_start_epoch=5

SSD快速迁移学习训练日志与文件生成与SSD基本检测相似,请参见表1

评估命令参考

SSD快速迁移学习模型评估的启动参考执行以下命令。本命令与SSD检测评估命令的参数一致,请参见表2

python3 model_eval.py --eval_dataset_path={eval_dataset_path} --eval_ckpt_path=./output_dir --eval_output_path=./eval_result --min_score=0.01 --nms_threshold=0.2 --pretrained_ckpt_path= pre_trained_ckpt