Refine DRF preprocessing and body-prior pipeline
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
data_cfg:
|
||||
dataset_name: Scoliosis1K
|
||||
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118
|
||||
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-paper
|
||||
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
|
||||
num_workers: 1
|
||||
remove_no_gallery: false
|
||||
@@ -10,7 +10,7 @@ evaluator_cfg:
|
||||
enable_float16: true
|
||||
restore_ckpt_strict: true
|
||||
restore_hint: 20000
|
||||
save_name: DRF
|
||||
save_name: DRF_paper
|
||||
eval_func: evaluate_scoliosis
|
||||
sampler:
|
||||
batch_shuffle: false
|
||||
@@ -19,7 +19,7 @@ evaluator_cfg:
|
||||
frames_all_limit: 720
|
||||
metric: euc
|
||||
transform:
|
||||
- type: BaseSilCuttingTransform
|
||||
- type: BaseSilTransform
|
||||
- type: NoOperation
|
||||
|
||||
loss_cfg:
|
||||
@@ -90,7 +90,7 @@ trainer_cfg:
|
||||
restore_ckpt_strict: true
|
||||
restore_hint: 0
|
||||
save_iter: 20000
|
||||
save_name: DRF
|
||||
save_name: DRF_paper
|
||||
sync_BN: true
|
||||
total_iter: 20000
|
||||
sampler:
|
||||
@@ -102,5 +102,5 @@ trainer_cfg:
|
||||
sample_type: fixed_unordered
|
||||
type: TripletSampler
|
||||
transform:
|
||||
- type: BaseSilCuttingTransform
|
||||
- type: BaseSilTransform
|
||||
- type: NoOperation
|
||||
|
||||
Reference in New Issue
Block a user