# LatteArt Judge

主要功能可以讓用戶在放入拉花的照片後，自動辨別拉花圖形的區域並切割照片。接著再將切割出的拉花照片，經由模型評估拉花複雜度後打分

### 使用方法
首先打開cmd，安裝所需模組

pip install -r requirements.txt

1. 下載模型權重檔，並放在".\EFN_Model"這個目錄中，(檔案大，約500MB)
下載連結：https://drive.google.com/file/d/1UJ1s22IaKd3dfBMo9ACP0RNJj2OFHb-J/view?usp=sharing
放在此處

2. 將輸入照片放入目錄".\main\inputPhoto"中，接著執行.\main\main.py

3. 裁切好的拉花照片會被儲存為".\main\cropPhoto\test.jpg"

4. 分數輸出結果將會顯示在terminal

## Reference
### 常用指令

yolov5 訓練模型指令 (建議縮小batch到16，避免記憶體不足。img_size使用64即可)

python .\yolov5\train.py --img-size 10 --batch 128 --epochs 100 --data latte.yaml --weights best.pt

可用tensorboard監控yolov5訓練過程

tensorboard --logdir yolov5\runs\train

---

### 常見問題

yolov5模型訓練教學網站
https://officeguide.cc/pytorch-yolo-v5-object-egg-detection-models-tutorial-examples/#google_vignette

yolov5資料集標註工具閃退問題
https://blog.csdn.net/m0_74232237/article/details/130985914