13 lines
337 B
Python
Executable File
13 lines
337 B
Python
Executable File
import os
|
|
|
|
# Cấu hình thư mục lưu trữ ảnh
|
|
IMAGE_FOLDER = "images"
|
|
LABEL_FOLDER = "label"
|
|
TEMP_FOLDER = "temp"
|
|
TRAINED_MODEL_FOLDER = "trained_models"
|
|
PRETRAINED_MODEL = "train5/weights/best.pt"
|
|
|
|
# Tạo thư mục nếu chưa có
|
|
for folder in [IMAGE_FOLDER, LABEL_FOLDER, TEMP_FOLDER]:
|
|
os.makedirs(folder, exist_ok=True)
|