66 lines
1.3 KiB
YAML
66 lines
1.3 KiB
YAML
data_cfg:
|
|
dataset_name: Scoliosis1K
|
|
data_type: heatmap
|
|
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl
|
|
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
|
|
num_workers: 1
|
|
data_in_use: [true, true]
|
|
remove_no_gallery: false
|
|
test_dataset_name: Scoliosis1K
|
|
|
|
evaluator_cfg:
|
|
enable_float16: true
|
|
restore_ckpt_strict: true
|
|
restore_hint: ./artifact/scoliosis_drf_author_118_compat/DRF_118_unordered_iter2w_lr0.001_8830-08000.pt
|
|
save_name: DRF_author_eval_118_1gpu
|
|
eval_func: evaluate_scoliosis
|
|
sampler:
|
|
batch_shuffle: false
|
|
batch_size: 1
|
|
sample_type: all_ordered
|
|
type: InferenceSampler
|
|
frames_all_limit: 720
|
|
metric: euc
|
|
transform:
|
|
- type: BaseSilCuttingTransform
|
|
- type: NoOperation
|
|
|
|
model_cfg:
|
|
model: DRF
|
|
label_order:
|
|
- negative
|
|
- positive
|
|
- neutral
|
|
num_pairs: 8
|
|
num_metrics: 3
|
|
backbone_cfg:
|
|
type: ResNet9
|
|
block: BasicBlock
|
|
in_channel: 2
|
|
channels:
|
|
- 64
|
|
- 128
|
|
- 256
|
|
- 512
|
|
layers:
|
|
- 1
|
|
- 1
|
|
- 1
|
|
- 1
|
|
strides:
|
|
- 1
|
|
- 2
|
|
- 2
|
|
- 1
|
|
maxpool: false
|
|
SeparateFCs:
|
|
in_channels: 512
|
|
out_channels: 256
|
|
parts_num: 16
|
|
SeparateBNNecks:
|
|
class_num: 3
|
|
in_channels: 256
|
|
parts_num: 16
|
|
bin_num:
|
|
- 16
|