fix
parent
aae8449532
commit
ebe19ce844
|
@ -48,8 +48,8 @@ DataLoader:
|
||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
|
cls_label_path: ./dataset/safety_helmet/train_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
@ -77,8 +77,8 @@ DataLoader:
|
||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
|
cls_label_path: ./dataset/safety_helmet/val_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
|
|
@ -48,8 +48,8 @@ DataLoader:
|
||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
|
cls_label_path: ./dataset/safety_helmet/train_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
@ -76,8 +76,8 @@ DataLoader:
|
||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
|
cls_label_path: ./dataset/safety_helmet/val_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
|
|
@ -86,8 +86,8 @@ DataLoader:
|
||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
|
cls_label_path: ./dataset/safety_helmet/train_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
@ -126,8 +126,8 @@ DataLoader:
|
||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
|
cls_label_path: ./dataset/safety_helmet/val_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
|
|
@ -49,8 +49,8 @@ DataLoader:
|
||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
|
cls_label_path: ./dataset/safety_helmet/train_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
@ -90,8 +90,8 @@ DataLoader:
|
||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
|
cls_label_path: ./dataset/safety_helmet/val_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
|
|
@ -7,7 +7,7 @@ Global:
|
||||||
save_interval: 1
|
save_interval: 1
|
||||||
eval_during_train: True
|
eval_during_train: True
|
||||||
eval_interval: 1
|
eval_interval: 1
|
||||||
epochs: 60
|
epochs: 40
|
||||||
print_batch_step: 10
|
print_batch_step: 10
|
||||||
use_visualdl: False
|
use_visualdl: False
|
||||||
# used for static mode and model export
|
# used for static mode and model export
|
||||||
|
@ -47,8 +47,8 @@ DataLoader:
|
||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
|
cls_label_path: ./dataset/safety_helmet/train_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
@ -78,8 +78,8 @@ DataLoader:
|
||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
|
cls_label_path: ./dataset/safety_helmet/val_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
|
|
@ -52,8 +52,8 @@ DataLoader:
|
||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/train_list.txt
|
cls_label_path: ./dataset/safety_helmet/train_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
@ -94,8 +94,8 @@ DataLoader:
|
||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: ImageNetDataset
|
name: ImageNetDataset
|
||||||
image_root: ./dataset/SafetyHelmetV3/
|
image_root: ./dataset/safety_helmet/
|
||||||
cls_label_path: ./dataset/SafetyHelmetV3/eval_list.txt
|
cls_label_path: ./dataset/safety_helmet/val_list.txt
|
||||||
transform_ops:
|
transform_ops:
|
||||||
- DecodeImage:
|
- DecodeImage:
|
||||||
to_rgb: True
|
to_rgb: True
|
||||||
|
|
Loading…
Reference in New Issue