21 lines
691 B
YAML
21 lines
691 B
YAML
# Train on human YOLO labels, preserving the existing fixed-tile train/val split.
|
|
_base_: film.yaml
|
|
work_dir: runs/classify_tiles128_resnet18
|
|
data:
|
|
root: data/film_yolo_128
|
|
format: yolo_tiles
|
|
yaml: data/film_yolo_128/data.yaml
|
|
image_size: [128, 128]
|
|
classes: [black, blackdot, grease, mosquito, normal, obstacle, ripple, uneven, wrinkle]
|
|
train_transforms:
|
|
- {name: RandomHorizontalFlip, p: 0.5}
|
|
- {name: RandomVerticalFlip, p: 0.5}
|
|
- {name: ToTensor}
|
|
val_transforms:
|
|
- {name: ToTensor}
|
|
train:
|
|
# Strong inverse-frequency weights plus uniform smoothing bias scores toward rare classes.
|
|
label_smoothing: 0.0
|
|
swanlab:
|
|
tags: [classify, resnet18, tiles128]
|