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

fix

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