Spaces:
Running
Running
File size: 305 Bytes
fb59b22 4ecf9fd 0bb457d 4ecf9fd fb59b22 |
1 2 3 4 5 6 7 8 9 |
EPOCH = 500
DEFAULT_IMAGE_SIZE = 640
BATCH = 10
RESUME_TRAIN = true
RECREATE_DATASET = true
YOLO_BASE_MODEL_NAME = "yolo11s-seg"
YOLO_MODEL_NAME = "final_model_yolo11s-seg"
IMAGE_SOURCE_PATH = "images"
YOLO_MODEL_REMOTE_URL = "https://huggingface.co/mosesb/best-comic-panel-detection/resolve/main/best.pt" |