From 471698f5a490ff50c6f4de7d9f5a71959601603c Mon Sep 17 00:00:00 2001 From: leo Date: Thu, 7 Mar 2024 19:40:34 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=BB=BA=E5=88=97=20ann=5Ffile=5Flbs?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bak/init_data.py | 1 + db_utils.py | 1 + 2 files changed, 2 insertions(+) diff --git a/bak/init_data.py b/bak/init_data.py index 477e772..9590782 100644 --- a/bak/init_data.py +++ b/bak/init_data.py @@ -12,6 +12,7 @@ class BatchDataBase(BaseModel): is_validation: bool | None = None ann_file: str | None = None + ann_file_lbs: str | None = None img_prefix: str | None = None filter_empty_gt: bool | None = None update_cache: bool | None = None diff --git a/db_utils.py b/db_utils.py index 4afde3b..ff1fd48 100644 --- a/db_utils.py +++ b/db_utils.py @@ -19,6 +19,7 @@ class BatchData(Base): is_train = Column(Integer) is_validation = Column(Integer) ann_file = Column(String) + ann_file_lbs = Column(String) img_prefix = Column(String) filter_empty_gt = Column(Integer) update_cache = Column(Integer)