Files
khqp_trainer/bak/init_data.py
2024-03-07 19:40:34 +08:00

28 lines
621 B
Python

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
from pydantic import BaseModel
class BatchDataBase(BaseModel):
year: int | None = None
census_batch: str | None = None # 普查批次
id_code: str | None = None # 编号
precision: str | None = None # 精度
is_train: bool | None = None
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
class BatchDataRead(BatchDataBase):
id: int
class Config:
orm_mode = True