17 lines
380 B
YAML
17 lines
380 B
YAML
# Lightweight nine-class classifier; preserve the ResNet18 preprocessing/split.
|
|
_base_: film.yaml
|
|
|
|
work_dir: runs/classify_mobilenet_v3_small
|
|
|
|
data:
|
|
root: data/orgnized_from_r101
|
|
classes: [black, blackdot, grease, mosquito, normal, obstacle, ripple, uneven, wrinkle]
|
|
|
|
model:
|
|
backbone:
|
|
arch: mobilenet_v3_small
|
|
|
|
train:
|
|
swanlab:
|
|
tags: [classify, mobilenet_v3_small]
|