Dipake kalau mau ngetes inference aja
Complete contents for assets/bromelia folder is here:
Google Drive
Dataset sementara kunci dulu pakai oxford_flowers102. distillation.py otomatis ngambil teacher ke generated/oxford_flowers102.
-
generator.py
--dataset
: nama dataset--module-name
: preprocessing pakai modul apa, ini sesuaikan sama model yg mau dipakai.--model-name
: model yang mau di generate--image-size
: ukuran imagenya mau sekian kali sekian (kalau isi 224, berarti 224 x 224)--run-num
: angka running. (1, 2, 3... etc.)
Script:
python3 generator.py --dataset oxford_flowers102 --module-name resnet --model-name ResNet101 --image-size 224 --run-num 1
-
distillation.py
--module-name
: preprocessing buat student modelnya.--student-model
: nama student model--teacher
: nama teacher (teacher harus ada dulu di folder generator/oxford_flowers102)--run-num
: angka running. (1, 2, 3... etc.) 1
Script:
python3 distillation.py --module-name mobilenet_v3 --student-model MobileNetV3Small --teacher ResNet101 --run-num 1