From a3e7e29897c9cc3a7cc5e9284906c2bc22f7c282 Mon Sep 17 00:00:00 2001
From: nguentrungthat <entjtj2000@gmail.com>
Date: Tue, 25 Mar 2025 13:46:23 +0700
Subject: [PATCH] Update train.py

---
 src/server/train.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/server/train.py b/src/server/train.py
index 43689f5..7abe497 100755
--- a/src/server/train.py
+++ b/src/server/train.py
@@ -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)