taichunyen/gotoubun-no-discriminator

五等分的花嫁辨識器

五等分辨識器

動機

使用工具

• Pytorch
• PyQt5

file overview

• train.py [parameter]
  parameter list:
  --n_epochs
  --batch_size
  --n_classes
  --dataroot
  --testroot

• test.py [parameter]
  parameter list:
  --model(trained model dir)
  --n_classes
  --root(test_img dir)

• train_on_colab.ipynb(use train.py on colab)
  (if vram out of memory use train_alex.py & test_alex.py or train_on_colab.ipynb)

• gotoubun_qt/PyGui.py(run the GUI)

• gotoubun_qt/myui.py(GUI sturcture)

• dataset/(training data)

• gotoubun/(testing data)

訓練模型

• resnext50_32x4d
• AlexNet

訓練過程

accuracy(resnext50_32x4d):

loss(resnext50_32x4d):

GUI(pyqt5)

結果

python3.6 ./gotoubun_qt/PyGui.py

誌謝

感謝凶宸提供data來源！