From 219cf23abc375d9170aec941d0d5ba5411862f51 Mon Sep 17 00:00:00 2001 From: Shaun Zhou Date: Wed, 14 Oct 2026 13:29:58 +0800 Subject: [PATCH] feat: add MobileNetV3-Small classifier configuration --- configs/classify/film_mobilenet_v3_small.yaml | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 configs/classify/film_mobilenet_v3_small.yaml 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]