EasyData/deploy/configs/ppdataaug_shitu.yaml at develop · andyjiang1116/EasyData · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
DataGen:
  ops:
    - randaugment
    - random_erasing
    - gridmask
    - tia_distort
    - tia_stretch
    - tia_perspective
  ori_data_dir: "demo/shitu_data"
  label_file: "demo/shitu_data/train_list.txt"
  gen_label: &gen_label "labels/test.txt"
  img_save_folder: &img_save_folder "test"
  gen_ratio: 0
  gen_num: 5
  size: 224

FeatureExtract:
  config: "deploy/configs/ppcv/feature_extract.yaml"
  thresh: 0.9
  file_out: "tmp/rm_repeat.txt"

  # indexing engine config
IndexProcess:
  index_method: "HNSW32" # supported: HNSW32, IVF, Flat
  image_root: *img_save_folder
  index_dir: "./augdata/all_aug"
  all_label_file:  *gen_label
  index_operation: "new" # suported: "append", "remove", "new"
  delimiter: " "
  dist_type: "IP"
  embedding_size: 512
  batch_size: 32
  return_k: 5
  score_thres: 0.5