yolo数据集标记

2025-12-25 17:41:18 +08:00
parent 1a135cdda7
commit a0555ac6b5
22 changed files with 10194 additions and 1 deletions
@@ -0,0 +1,39 @@
+import os
+import sys
+
+if getattr(sys, 'frozen', False):
+    BASE_DIR = os.path.dirname(sys.executable)
+else:
+    BASE_DIR = os.path.abspath(os.path.dirname(__file__))
+
+DATABASE_FILE = os.path.join(BASE_DIR, 'ftc_ml.db')
+STORAGE_DIR = os.path.join(BASE_DIR, 'local_storage')
+PROTOTYPE_FILE = os.path.join(STORAGE_DIR, 'prototype_library.pt')
+PREPROCESSED_CACHE_FILE = os.path.join(STORAGE_DIR, 'preprocessed_cache.pt')
+
+
+MAX_DESCRIPTION_LENGTH = 30
+MAX_VIDEO_SIZE_MB = 10000
+MAX_VIDEO_SIZE_BYTES = MAX_VIDEO_SIZE_MB * 1000 * 1000
+MAX_VIDEO_LENGTH_SECONDS = 120
+MAX_FRAMES_PER_VIDEO = 10000
+MAX_VIDEO_RESOLUTION_WIDTH = 3840
+MAX_VIDEO_RESOLUTION_HEIGHT = 2160
+MAX_DATASETS_PER_TEAM = 50
+MAX_VIDEOS_PER_TEAM = 50
+TRACKER_FNS = [
+    'CSRT', 'MedianFlow', 'MIL', 'MOSSE', 'TLD', 'KCF', 'Boosting',
+]
+
+def get_limit_data_for_render_template():
+    return {
+        'MAX_VIDEO_SIZE_BYTES': MAX_VIDEO_SIZE_BYTES,
+        'MAX_VIDEO_SIZE_MB': MAX_VIDEO_SIZE_MB,
+        'MAX_VIDEO_LENGTH_SECONDS': MAX_VIDEO_LENGTH_SECONDS,
+        'MAX_FRAMES_PER_VIDEO': MAX_FRAMES_PER_VIDEO,
+        'MAX_DESCRIPTION_LENGTH': MAX_DESCRIPTION_LENGTH,
+    }
+
+ONNX_MODELS_DIR = "onnx_models"
+MOBILENET_LARGE_ONNX = os.path.join(ONNX_MODELS_DIR, 'mobilenet_v3_large.onnx')
+MOBILENET_SMALL_ONNX = os.path.join(ONNX_MODELS_DIR, 'mobilenet_v3_small.onnx')