pull/1974/head
gaotingquan 2022-06-09 03:41:43 +00:00
parent aae8449532
commit ebe19ce844
No known key found for this signature in database
GPG Key ID: F3EF7F42536A30B7
6 changed files with 25 additions and 25 deletions

View File

@ -48,8 +48,8 @@ DataLoader:
Train:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/train_list.txt
transform_ops:
- DecodeImage:
to_rgb: True
@ -77,8 +77,8 @@ DataLoader:
Eval:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/val_list.txt
transform_ops:
- DecodeImage:
to_rgb: True

View File

@ -48,8 +48,8 @@ DataLoader:
Train:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/train_list.txt
transform_ops:
- DecodeImage:
to_rgb: True
@ -76,8 +76,8 @@ DataLoader:
Eval:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/val_list.txt
transform_ops:
- DecodeImage:
to_rgb: True

View File

@ -86,8 +86,8 @@ DataLoader:
Train:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/train_list.txt
transform_ops:
- DecodeImage:
to_rgb: True
@ -126,8 +126,8 @@ DataLoader:
Eval:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/val_list.txt
transform_ops:
- DecodeImage:
to_rgb: True

View File

@ -49,8 +49,8 @@ DataLoader:
Train:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/train_list.txt
transform_ops:
- DecodeImage:
to_rgb: True
@ -90,8 +90,8 @@ DataLoader:
Eval:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/val_list.txt
transform_ops:
- DecodeImage:
to_rgb: True

View File

@ -7,7 +7,7 @@ Global:
save_interval: 1
eval_during_train: True
eval_interval: 1
epochs: 60
epochs: 40
print_batch_step: 10
use_visualdl: False
# used for static mode and model export
@ -47,8 +47,8 @@ DataLoader:
Train:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/train_list.txt
transform_ops:
- DecodeImage:
to_rgb: True
@ -78,8 +78,8 @@ DataLoader:
Eval:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/val_list.txt
transform_ops:
- DecodeImage:
to_rgb: True

View File

@ -52,8 +52,8 @@ DataLoader:
Train:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/train_list.txt
transform_ops:
- DecodeImage:
to_rgb: True
@ -94,8 +94,8 @@ DataLoader:
Eval:
dataset:
name: ImageNetDataset
image_root: ./dataset/SafetyHelmetV3/
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
image_root: ./dataset/safety_helmet/
cls_label_path: ./dataset/safety_helmet/val_list.txt
transform_ops:
- DecodeImage:
to_rgb: True