diff --git a/README.md b/README.md index 48b95f2..bb87921 100644 --- a/README.md +++ b/README.md @@ -29,12 +29,4 @@ python main.py [安装链接](https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/windows-pip.html) -2. 模型文件连接 - - 由于软件包含已训练好的三个模型,模型参数较大,放在百度网盘自行下载,并放在根目录。 - - 链接:https://pan.baidu.com/s/1Wf4gjmccgqQYeknm9pX2jQ - - 提取码:2gs7 - -3. 新模型文件自动更新在源码中,pic中为测试图片 \ No newline at end of file +2. 模型文件全部更新在inference_model中,pic为测试图片 diff --git a/inference_model/ppyolov2_r50vd_dcn_voc/infer_cfg.yml b/inference_model/ppyolov2_r50vd_dcn_voc/infer_cfg.yml new file mode 100644 index 0000000..ff48818 --- /dev/null +++ b/inference_model/ppyolov2_r50vd_dcn_voc/infer_cfg.yml @@ -0,0 +1,27 @@ +mode: fluid +draw_threshold: 0.5 +metric: VOC +arch: YOLO +min_subgraph_size: 3 +Preprocess: +- interp: 2 + keep_ratio: false + target_size: + - 640 + - 640 + type: Resize +- is_scale: true + mean: + - 0.485 + - 0.456 + - 0.406 + std: + - 0.229 + - 0.224 + - 0.225 + type: NormalizeImage +- type: Permute +label_list: +- flv +- gx +- mbw diff --git a/inference_model/ppyolov2_r50vd_dcn_voc/model.pdiparams b/inference_model/ppyolov2_r50vd_dcn_voc/model.pdiparams new file mode 100644 index 0000000..e1ab8f0 Binary files /dev/null and b/inference_model/ppyolov2_r50vd_dcn_voc/model.pdiparams differ diff --git a/inference_model/ppyolov2_r50vd_dcn_voc/model.pdiparams.info b/inference_model/ppyolov2_r50vd_dcn_voc/model.pdiparams.info new file mode 100644 index 0000000..d8129c0 Binary files /dev/null and b/inference_model/ppyolov2_r50vd_dcn_voc/model.pdiparams.info differ diff --git a/inference_model/ppyolov2_r50vd_dcn_voc/model.pdmodel b/inference_model/ppyolov2_r50vd_dcn_voc/model.pdmodel new file mode 100644 index 0000000..ab290a2 Binary files /dev/null and b/inference_model/ppyolov2_r50vd_dcn_voc/model.pdmodel differ diff --git a/inference_model/yolov3/infer_cfg.yml b/inference_model/yolov3/infer_cfg.yml new file mode 100644 index 0000000..a2a783b --- /dev/null +++ b/inference_model/yolov3/infer_cfg.yml @@ -0,0 +1,29 @@ +mode: fluid +draw_threshold: 0.5 +metric: VOC +use_dynamic_shape: false +arch: YOLO +min_subgraph_size: 3 +Preprocess: +- interp: 2 + keep_ratio: false + target_size: + - 608 + - 608 + type: Resize +- is_scale: true + mean: + - 0.485 + - 0.456 + - 0.406 + std: + - 0.229 + - 0.224 + - 0.225 + type: NormalizeImage +- type: Permute +label_list: +- flv +- gx +- mbw +- object diff --git a/inference_model/yolov3/model.pdiparams b/inference_model/yolov3/model.pdiparams new file mode 100644 index 0000000..167ac50 Binary files /dev/null and b/inference_model/yolov3/model.pdiparams differ diff --git a/inference_model/yolov3/model.pdiparams.info b/inference_model/yolov3/model.pdiparams.info new file mode 100644 index 0000000..72c4dc6 Binary files /dev/null and b/inference_model/yolov3/model.pdiparams.info differ diff --git a/inference_model/yolov3/model.pdmodel b/inference_model/yolov3/model.pdmodel new file mode 100644 index 0000000..ebc0065 Binary files /dev/null and b/inference_model/yolov3/model.pdmodel differ