|
|
@@ -1,8 +1,8 @@ |
|
|
|
# global configs |
|
|
|
Global: |
|
|
|
checkpoints: null |
|
|
|
pretrained_model: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/PPShiTuV2/general_PPLCNetV2_base_pretrained_v1.0.pdparams |
|
|
|
output_dir: /tmp/output/ |
|
|
|
pretrained_model: /root/.paddleclas/weights/general_PPLCNetV2_base_pretrained_v1.0.pdparams |
|
|
|
output_dir: /model/ |
|
|
|
device: gpu |
|
|
|
save_interval: 1 |
|
|
|
eval_during_train: True |
|
|
@@ -16,7 +16,7 @@ Global: |
|
|
|
use_dali: False |
|
|
|
# used for static mode and model export |
|
|
|
image_shape: [3, 224, 224] |
|
|
|
save_inference_dir: /tmp/output/inference |
|
|
|
save_inference_dir: /model/inference |
|
|
|
|
|
|
|
# AMP: |
|
|
|
# scale_loss: 65536 |
|
|
@@ -95,8 +95,8 @@ DataLoader: |
|
|
|
Train: |
|
|
|
dataset: |
|
|
|
name: ImageNetDataset |
|
|
|
image_root: /tmp/dataset/ |
|
|
|
cls_label_path: /tmp/dataset/train_list.txt |
|
|
|
image_root: /dataset/ |
|
|
|
cls_label_path: /dataset/train_list.txt |
|
|
|
relabel: True |
|
|
|
transform_ops: |
|
|
|
- DecodeImage: |
|
|
@@ -145,8 +145,8 @@ DataLoader: |
|
|
|
Query: |
|
|
|
dataset: |
|
|
|
name: VeriWild |
|
|
|
image_root: /tmp/dataset/ |
|
|
|
cls_label_path: /tmp/dataset/val_list.txt |
|
|
|
image_root: /dataset/ |
|
|
|
cls_label_path: /dataset/val_list.txt |
|
|
|
transform_ops: |
|
|
|
- DecodeImage: |
|
|
|
to_rgb: True |
|
|
@@ -173,8 +173,8 @@ DataLoader: |
|
|
|
Gallery: |
|
|
|
dataset: |
|
|
|
name: VeriWild |
|
|
|
image_root: /tmp/dataset/ |
|
|
|
cls_label_path: /tmp/dataset/val_list.txt |
|
|
|
image_root: /dataset/ |
|
|
|
cls_label_path: /dataset/val_list.txt |
|
|
|
transform_ops: |
|
|
|
- DecodeImage: |
|
|
|
to_rgb: True |
|
|
|