andrew.ng
/
YoLo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update train.py

This commit is contained in:
nguentrungthat 2025-03-25 13:46:23 +07:00
parent 052e385473
commit a3e7e29897
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/server/train.py
View File

@ -13,6 +13,7 @@ import sqlite3
DATA_SPLIT_FOLDER = "model_datasets"
IMAGE_EXTENSION = ".png"
LOG_FILE = "training_logs.log"
LATEST_DETECT = "latest_detect"
LATEST_DETECT_IMAGES = "latest_detect/images"
LATEST_DETECT_LABELS = "latest_detect/labels"
image_folder=""
@ -271,8 +272,8 @@ def train_yolo_model(
call_reload_model_api()
clear_images_source()
if(len(label_folder) >= 20):
copy_file_latest_training()
if(len(label_files) >= 20):
if os.path.exists(LATEST_DETECT) copy_file_latest_training()
# 🚀 Copy files
train_copied_imgs, train_copied_lbls = move_files(train_files, train_img_folder, train_lbl_folder)