leo/khqp_trainer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

新建列 ann_file_lbs

Browse Source
This commit is contained in:
leo
2024-03-07 19:40:34 +08:00
parent 30d0659c03
commit 471698f5a4
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
bak/init_data.py
View File

@@ -12,6 +12,7 @@ class BatchDataBase(BaseModel):
is_validation: bool | None = None is_validation: bool | None = None
ann_file: str | None = None ann_file: str | None = None
ann_file_lbs: str | None = None
img_prefix: str | None = None img_prefix: str | None = None
filter_empty_gt: bool | None = None filter_empty_gt: bool | None = None
update_cache: bool | None = None update_cache: bool | None = None

1
db_utils.py
View File

@@ -19,6 +19,7 @@ class BatchData(Base):
is_train = Column(Integer) is_train = Column(Integer)
is_validation = Column(Integer) is_validation = Column(Integer)
ann_file = Column(String) ann_file = Column(String)
ann_file_lbs = Column(String)
img_prefix = Column(String) img_prefix = Column(String)
filter_empty_gt = Column(Integer) filter_empty_gt = Column(Integer)
update_cache = Column(Integer) update_cache = Column(Integer)