未验证 提交 ebe19ce8 编写于 作者: G gaotingquan

fix

上级 aae84495
......@@ -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
......
......@@ -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
......
......@@ -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
......
......@@ -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
......
......@@ -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
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册