feat: add MobileNetV3-Small classifier configuration

2026-10-14 13:29:58 +08:00
parent 03f667276b
commit 219cf23abc
1 changed files with 16 additions and 0 deletions
@@ -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]