diff --git a/configs/classify/film_mobilenet_v3_small.yaml b/configs/classify/film_mobilenet_v3_small.yaml new file mode 100644 index 0000000..b3ddcac --- /dev/null +++ b/configs/classify/film_mobilenet_v3_small.yaml @@ -0,0 +1,16 @@ +# 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]