cuicheng01
GitHub
commit
ab45ca9d17
deploy
configs/PULC
language_classification
text_image_orientation
textline_orientation
vehicle_exists
images/PULC/vehicle_exists
ppcls
configs/PULC
|
|
@ -28,6 +28,6 @@ PostProcess:
|
|||
main_indicator: Topk
|
||||
Topk:
|
||||
topk: 2
|
||||
class_id_map_file: "../dataset/language_classification/label_list.txt"
|
||||
class_id_map_file: "../ppcls/utils/PULC_label_list/language_classification_label_list.txt"
|
||||
SavePreLabel:
|
||||
save_dir: ./pre_label/
|
||||
|
|
|
|||
|
|
@ -30,6 +30,6 @@ PostProcess:
|
|||
main_indicator: Topk
|
||||
Topk:
|
||||
topk: 2
|
||||
class_id_map_file: "../dataset/text_image_orientation/label_list.txt"
|
||||
class_id_map_file: "../ppcls/utils/PULC_label_list/text_image_orientation_label_list.txt"
|
||||
SavePreLabel:
|
||||
save_dir: ./pre_label/
|
||||
|
|
|
|||
|
|
@ -28,6 +28,6 @@ PostProcess:
|
|||
main_indicator: Topk
|
||||
Topk:
|
||||
topk: 1
|
||||
class_id_map_file: "../ppcls/utils/PULC/textline_orientation_label_list.txt"
|
||||
class_id_map_file: "../ppcls/utils/PULC_label_list/textline_orientation_label_list.txt"
|
||||
SavePreLabel:
|
||||
save_dir: ./pre_label/
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
Global:
|
||||
infer_imgs: "./images/PULC/car_exists/objects365_00001507.jpeg"
|
||||
inference_model_dir: "./models/car_exists_infer"
|
||||
infer_imgs: "./images/PULC/vehicle_exists/objects365_00001507.jpeg"
|
||||
inference_model_dir: "./models/vehicle_exists_infer"
|
||||
batch_size: 1
|
||||
use_gpu: True
|
||||
enable_mkldnn: False
|
||||
|
|
@ -30,7 +30,7 @@ PostProcess:
|
|||
main_indicator: ThreshOutput
|
||||
ThreshOutput:
|
||||
threshold: 0.5
|
||||
label_0: nocar
|
||||
label_1: contains_car
|
||||
label_0: no_vehicle
|
||||
label_1: contains_vehicle
|
||||
SavePreLabel:
|
||||
save_dir: ./pre_label/
|
||||
|
Before Width: | Height: | Size: 157 KiB After Width: | Height: | Size: 157 KiB |
|
Before Width: | Height: | Size: 178 KiB After Width: | Height: | Size: 178 KiB |
|
|
@ -1,12 +1,12 @@
|
|||
# PULC语种分类模型
|
||||
# PULC 语种分类模型
|
||||
|
||||
## 目录
|
||||
|
||||
- [1. 模型和应用场景介绍](#1)
|
||||
- [2. 模型快速体验](#2)
|
||||
- [1. 模型和应用场景介绍](#1)
|
||||
- [2. 模型快速体验](#2)
|
||||
- [2.1 安装 paddleclas](#2.1)
|
||||
- [2.2 预测](#2.2)
|
||||
- [3. 模型训练、评估和预测](#3)
|
||||
- [3. 模型训练、评估和预测](#3)
|
||||
- [3.1 环境配置](#3.1)
|
||||
- [3.2 数据准备](#3.2)
|
||||
- [3.2.1 数据集来源](#3.2.1)
|
||||
|
|
@ -73,14 +73,14 @@ pip3 install paddleclas
|
|||
- 使用命令行快速预测
|
||||
|
||||
```
|
||||
paddleclas --model_name=language_classification --infer_imgs=deploy/images/PULC/language_classification/img_rot0_demo.jpg
|
||||
paddleclas --model_name=language_classification --infer_imgs=deploy/images/PULC/language_classification/word_35404.png
|
||||
```
|
||||
|
||||
结果如下：
|
||||
|
||||
```
|
||||
>>> result
|
||||
class_ids: [4, 9], scores: [0.96809, 0.01001], label_names: ['japan', 'latin'], filename: deploy/images/PULC/language_classification/word_35404.png
|
||||
class_ids: [4, 6], scores: [0.88672, 0.01434], label_names: ['japan', 'korean'], filename: deploy/images/PULC/language_classification/word_35404.png
|
||||
Predict complete!
|
||||
```
|
||||
|
||||
|
|
@ -99,7 +99,7 @@ print(next(result))
|
|||
|
||||
```
|
||||
>>> result
|
||||
[{'class_ids': [4, 9], 'scores': [0.96809, 0.01001], 'label_names': ['japan', 'latin'], 'filename': 'deploy/images/PULC/language_classification/word_35404.png'}]
|
||||
[{'class_ids': [4, 6], 'scores': [0.88672, 0.01434], 'label_names': ['japan', 'korean'], 'filename': '/deploy/images/PULC/language_classification/word_35404.png'}]
|
||||
```
|
||||
|
||||
<a name="3"></a>
|
||||
|
|
|
|||
|
|
@ -1,13 +1,12 @@
|
|||
# PULC含文字图像方向分类模型
|
||||
# PULC 含文字图像方向分类模型
|
||||
|
||||
## 目录
|
||||
|
||||
- [1. 模型和应用场景介绍](#1)
|
||||
- [2. 模型快速体验](#2)
|
||||
- [1. 模型和应用场景介绍](#1)
|
||||
- [2. 模型快速体验](#2)
|
||||
- [2.1 安装 paddleclas](#2.1)
|
||||
- [2.2 预测](#2.2)
|
||||
|
||||
- [3. 模型训练、评估和预测](#3)
|
||||
- [3. 模型训练、评估和预测](#3)
|
||||
- [3.1 环境配置](#3.1)
|
||||
- [3.2 数据准备](#3.2)
|
||||
- [3.2.1 数据集来源](#3.2.1)
|
||||
|
|
|
|||
|
|
@ -78,13 +78,13 @@ pip3 install paddlepaddle paddleclas
|
|||
* 使用命令行快速预测
|
||||
|
||||
```bash
|
||||
paddleclas --model_name traffic_sign --infer_imgs PaddleClas/deploy/images/PULC/traffic_sign/100999_83928.jpg
|
||||
paddleclas --model_name traffic_sign --infer_imgs deploy/images/PULC/traffic_sign/100999_83928.jpg
|
||||
```
|
||||
|
||||
结果如下：
|
||||
```
|
||||
>>> result
|
||||
class_ids: [182, 179, 162, 128, 24], scores: [0.98623, 0.01255, 0.00022, 0.00021, 0.00012], label_names: ['pl110', 'pl100', 'pl120', 'p26', 'pm10'], filename: PaddleClas/deploy/images/PULC/traffic_sign/100999_83928.jpg
|
||||
class_ids: [182, 179, 162, 128, 24], scores: [0.98623, 0.01255, 0.00022, 0.00021, 0.00012], label_names: ['pl110', 'pl100', 'pl120', 'p26', 'pm10'], filename: deploy/images/PULC/traffic_sign/100999_83928.jpg
|
||||
```
|
||||
|
||||
**备注**： 更换其他预测的数据时，只需要改变 `--infer_imgs=xx` 中的字段即可，支持传入整个文件夹。
|
||||
|
|
@ -94,7 +94,7 @@ class_ids: [182, 179, 162, 128, 24], scores: [0.98623, 0.01255, 0.00022, 0.00021
|
|||
```python
|
||||
import paddleclas
|
||||
model = paddleclas.PaddleClas(model_name="traffic_sign")
|
||||
result = model.predict(input_data="PaddleClas/deploy/images/PULC/traffic_sign/100999_83928.jpg")
|
||||
result = model.predict(input_data="deploy/images/PULC/traffic_sign/100999_83928.jpg")
|
||||
print(next(result))
|
||||
```
|
||||
|
||||
|
|
@ -102,7 +102,7 @@ print(next(result))
|
|||
|
||||
```
|
||||
result
|
||||
[{'class_ids': [182, 179, 162, 128, 24], 'scores': [0.98623, 0.01255, 0.00022, 0.00021, 0.00012], 'label_names': ['pl110', 'pl100', 'pl120', 'p26', 'pm10'], 'filename': 'PaddleClas/deploy/images/PULC/traffic_sign/100999_83928.jpg'}]
|
||||
[{'class_ids': [182, 179, 162, 128, 24], 'scores': [0.98623, 0.01255, 0.00022, 0.00021, 0.00012], 'label_names': ['pl110', 'pl100', 'pl120', 'p26', 'pm10'], 'filename': 'deploy/images/PULC/traffic_sign/100999_83928.jpg'}]
|
||||
```
|
||||
|
||||
<a name="3"></a>
|
||||
|
|
|
|||
|
|
@ -79,13 +79,13 @@ pip3 install paddlepaddle paddleclas
|
|||
* 使用命令行快速预测
|
||||
|
||||
```bash
|
||||
paddleclas --model_name vehicle_attribute --infer_imgs PaddleClas/deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg
|
||||
paddleclas --model_name vehicle_attribute --infer_imgs deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg
|
||||
```
|
||||
|
||||
结果如下：
|
||||
```
|
||||
>>> result
|
||||
attributes: Color: (yellow, prob: 0.9893476963043213), Type: (hatchback, prob: 0.9734097719192505), output: [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0], filename: PaddleClas/deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg
|
||||
attributes: Color: (yellow, prob: 0.9893476963043213), Type: (hatchback, prob: 0.9734097719192505), output: [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0], filename: deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg
|
||||
ppcls INFO: Predict complete!
|
||||
```
|
||||
|
||||
|
|
@ -96,7 +96,7 @@ ppcls INFO: Predict complete!
|
|||
```python
|
||||
import paddleclas
|
||||
model = paddleclas.PaddleClas(model_name="vehicle_attribute")
|
||||
result = model.predict(input_data="PaddleClas/deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg")
|
||||
result = model.predict(input_data="deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg")
|
||||
print(next(result))
|
||||
```
|
||||
|
||||
|
|
@ -104,7 +104,7 @@ print(next(result))
|
|||
|
||||
```
|
||||
result
|
||||
[{'attributes': 'Color: (yellow, prob: 0.9893476963043213), Type: (hatchback, prob: 0.9734097719192505)', 'output': [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0], 'filename': 'PaddleClas/deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg'}]
|
||||
[{'attributes': 'Color: (yellow, prob: 0.9893476963043213), Type: (hatchback, prob: 0.9734097719192505)', 'output': [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0], 'filename': 'deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg'}]
|
||||
```
|
||||
|
||||
<a name="3"></a>
|
||||
|
|
|
|||
|
|
@ -178,7 +178,7 @@ IMN_MODEL_SERIES = {
|
|||
PULC_MODEL_BASE_DOWNLOAD_URL = "https://paddleclas.bj.bcebos.com/models/PULC/{}_infer.tar"
|
||||
PULC_MODELS = [
|
||||
"person_exists", "person_attribute", "safety_helmet", "traffic_sign",
|
||||
"vehicle_exists", "vehicle_attr", "textline_orientation",
|
||||
"vehicle_exists", "vehicle_attribute", "textline_orientation",
|
||||
"text_image_orientation", "language_classification"
|
||||
]
|
||||
|
||||
|
|
@ -247,8 +247,10 @@ def init_config(model_type, model_name, inference_model_dir, **kwargs):
|
|||
cfg.PostProcess.Topk.class_id_map_file = kwargs[
|
||||
"class_id_map_file"]
|
||||
else:
|
||||
cfg.PostProcess.Topk.class_id_map_file = os.path.relpath(
|
||||
class_id_map_file_path = os.path.relpath(
|
||||
cfg.PostProcess.Topk.class_id_map_file, "../")
|
||||
cfg.PostProcess.Topk.class_id_map_file = os.path.join(
|
||||
__dir__, class_id_map_file_path)
|
||||
if "VehicleAttribute" in cfg.PostProcess:
|
||||
if "color_threshold" in kwargs and kwargs["color_threshold"]:
|
||||
cfg.PostProcess.VehicleAttribute.color_threshold = kwargs[
|
||||
|
|
|
|||
|
|
@ -121,7 +121,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/language_classification_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/language_classification_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -132,7 +132,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/language_classification_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/language_classification_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -152,7 +152,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/language_classification_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/language_classification_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
@ -161,4 +161,4 @@ Metric:
|
|||
topk: [1, 2]
|
||||
Eval:
|
||||
- TopkAcc:
|
||||
topk: [1, 2]
|
||||
topk: [1, 2]
|
||||
|
|
|
|||
|
|
@ -131,7 +131,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/language_classification_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/language_classification_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -152,7 +152,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/language_classification_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/language_classification_label_list.txt
|
||||
|
||||
Metric:
|
||||
Eval:
|
||||
|
|
|
|||
|
|
@ -121,7 +121,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/text_image_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/text_image_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -132,7 +132,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/text_image_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/text_image_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -152,7 +152,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/text_image_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/text_image_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
@ -161,4 +161,4 @@ Metric:
|
|||
topk: [1, 2]
|
||||
Eval:
|
||||
- TopkAcc:
|
||||
topk: [1, 2]
|
||||
topk: [1, 2]
|
||||
|
|
|
|||
|
|
@ -135,7 +135,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/text_image_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/text_image_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -149,7 +149,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 2
|
||||
class_id_map_file: ppcls/utils/PULC/text_image_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/text_image_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Eval:
|
||||
|
|
|
|||
|
|
@ -123,7 +123,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 1
|
||||
class_id_map_file: ppcls/utils/PULC/textline_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/textline_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -132,7 +132,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 1
|
||||
class_id_map_file: ppcls/utils/PULC/textline_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/textline_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -121,7 +121,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 1
|
||||
class_id_map_file: ppcls/utils/PULC/textline_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/textline_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -150,7 +150,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 1
|
||||
class_id_map_file: ppcls/utils/PULC/textline_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/textline_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -133,7 +133,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 1
|
||||
class_id_map_file: ppcls/utils/PULC/textline_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/textline_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -153,7 +153,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 1
|
||||
class_id_map_file: ppcls/utils/PULC/textline_orientation_label_list.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/textline_orientation_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -120,7 +120,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 5
|
||||
class_id_map_file: dataset/traffic_sign/label_name_id.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/traffic_sign_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -137,7 +137,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 5
|
||||
class_id_map_file: dataset/traffic_sign/label_name_id.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/traffic_sign_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -160,7 +160,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 5
|
||||
class_id_map_file: dataset/traffic_sign/label_name_id.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/traffic_sign_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -138,7 +138,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 5
|
||||
class_id_map_file: dataset/traffic_sign/label_name_id.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/traffic_sign_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -157,7 +157,7 @@ Infer:
|
|||
PostProcess:
|
||||
name: Topk
|
||||
topk: 5
|
||||
class_id_map_file: dataset/traffic_sign/label_name_id.txt
|
||||
class_id_map_file: ppcls/utils/PULC_label_list/traffic_sign_label_list.txt
|
||||
|
||||
Metric:
|
||||
Train:
|
||||
|
|
|
|||
|
|
@ -1,4 +0,0 @@
|
|||
0 0
|
||||
1 90
|
||||
2 180
|
||||
3 270
|
||||
Loading…
Reference in New Issue