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]