From 4742e29d215bc11c0b810176bcd8055cf5c4cf24 Mon Sep 17 00:00:00 2001 From: leo Date: Wed, 6 Mar 2024 23:11:39 +0800 Subject: [PATCH] =?UTF-8?q?=E5=9F=BA=E7=A1=80=E9=A1=B5=E9=9D=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 8 +++++ bak/batch_data_v1.csv | 71 ++++++++++++++++++++++++++++++++++++++++++ data.db | Bin 28672 -> 28672 bytes main.py | 56 +++++++++++++++++++-------------- 4 files changed, 112 insertions(+), 23 deletions(-) create mode 100644 bak/batch_data_v1.csv diff --git a/README.md b/README.md index 4ddfe8f..d4e188b 100644 --- a/README.md +++ b/README.md @@ -16,3 +16,11 @@ lbs_path 不生成 [streamlit_sqlite](https://docs.streamlit.io/library/advanced-features/connecting-to-data) + +模型训练的启动参数在这个页面配置, 包括训练集, 测试集, 会在这个页面编辑数据 +实际使用的数据是那两个path, +后续需要在这个页面补充训练数据或测试数据 + +每次运行都要把一些数据加载到数据库里, +还是说把这些数据处理到数据库里后, 之后就是对数据库操作了 +然后是不是 val_data 也需要放到这个数据库里 \ No newline at end of file diff --git a/bak/batch_data_v1.csv b/bak/batch_data_v1.csv new file mode 100644 index 0000000..c55f380 --- /dev/null +++ b/bak/batch_data_v1.csv @@ -0,0 +1,71 @@ +id,year,census_batch,id_code,precision,is_train,is_validation,ann_file,img_prefix,filter_empty_gt,update_cache +1,2022,KHDP,7flys,S3,1,0,/data/data/split_ss_yimu_m6_p2/train_khdp_7flys_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230531_p1p2,/data/data/split_ss_yimu_m6_p2/train_khdp_7flys_gsd5_orgsize_noDecay_seg-v2/images,0,0 +2,2022,KHDP,7flys,S3,1,0,/data/data/split_ss_yimu_m6_p1/train_khdp_v2_3flys_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230531_p1p2,/data/data/split_ss_yimu_m6_p1/train_khdp_v2_3flys_gsd5_orgsize_noDecay_seg-v2/images,0,0 +3,2022,KHDP,044,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_044_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_044_gsd5_orgsize_noDecay_seg-v2/images,0,0 +4,2022,KHDP,064,S3,1,1,/data/data/split_ss_yimu_m6_p4/test_khdp_064_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_064_gsd5_orgsize_noDecay_seg-v2/images,0,0 +5,2022,KHDP,065,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_065_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_065_gsd5_orgsize_noDecay_seg-v2/images,0,0 +6,2022,KHDP,073,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_073_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_073_gsd5_orgsize_noDecay_seg-v2/images,0,0 +7,2022,KHDP,072,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_072_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_072_gsd5_orgsize_noDecay_seg-v2/images,0,0 +8,2022,KHDP,013,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_013_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_013_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +9,2022,KHDP,015,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_015_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_015_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +10,2022,KHDP,016,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_016_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_016_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +11,2022,KHDP,028,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_028_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601,/data/data/split_ss_yimu_m6_p4/test_khdp_028_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +12,2022,KHDP,067,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_067_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_067_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +13,2022,KHDP,068,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_068_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_068_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +14,2022,KHDP,069,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_069_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_069_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +15,2022,KHDP,070,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_070_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_070_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +16,2022,KHDP,071,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_071_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_071_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +17,2022,KHDP,075,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_075_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_075_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +18,2022,KHDP,078,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_078_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_078_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +19,2022,KHDP,080,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_080_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_080_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +20,2022,KHDP,081,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_081_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_081_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +21,2022,KHDP,082,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_082_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_082_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +22,2022,KHDP,084,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_084_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_084_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +23,2022,KHDP,087,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_087_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_087_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +24,2022,KHDP,089,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_089_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_089_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +25,2022,KHDP,090,S3,1,0,/data/data/split_ss_yimu_m6_p4/test_khdp_090_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016,/data/data/split_ss_yimu_m6_p4/test_khdp_090_mod_loc_gsd5_orgsize_noDecay_seg-v2/images,0,0 +26,2023,KHXP,001a002,S3,1,0,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_001a002_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_001a002_gsd5_s2048/images,0,0 +27,2023,KHXP,003a004a006,S3,1,0,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_003a004a006_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_003a004a006_gsd5_s2048/images,0,0 +28,2023,KHXP,005,S3,1,0,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_005_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_005_gsd5_s2048/images,0,0 +29,2023,KHXP,007a008,S3,1,0,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_007a008_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_007a008_gsd5_s2048/images,0,0 +30,2023,KHQP,001a006a010,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_001a006a010_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_001a006a010_gsd5_s2048/images,0,0 +31,2023,KHQP,002a004,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_002a004_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_002a004_gsd5_s2048/images,0,0 +32,2023,KHQP,003,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_003_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_003_gsd5_s2048/images,0,0 +33,2023,KHQP,005,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_005_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_005_gsd5_s2048/images,0,0 +34,2023,KHQP,007a008a009,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_007a008a009_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_007a008a009_gsd5_s2048/images,0,0 +35,2023,KHQP,011,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_011_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_011_gsd5_s2048/images,0,0 +36,2023,KHQP,012a013,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_012a013_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_012a013_gsd5_s2048/images,0,0 +37,2023,KHQP,014a038,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_014a038_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_014a038_gsd5_s2048/images,0,0 +38,2023,KHQP,015a016,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_015a016_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_015a016_gsd5_s2048/images,0,0 +39,2023,KHQP,018a069,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_018a069_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_018a069_gsd5_s2048/images,0,0 +40,2023,KHQP,019a075,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_019a075_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_019a075_gsd5_s2048/images,0,0 +41,2023,KHQP,020a022,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_020a022_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_020a022_gsd5_s2048/images,0,0 +42,2023,KHQP,023a025,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_023a025_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_023a025_gsd5_s2048/images,0,0 +43,2023,KHQP,026a029a079,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_026a029a079_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_026a029a079_gsd5_s2048/images,0,0 +44,2023,KHQP,028a027a030,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_028a027a030_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_028a027a030_gsd5_s2048/images,0,0 +45,2023,KHQP,032a033,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_032a033_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_032a033_gsd5_s2048/images,0,0 +46,2023,KHQP,039a040,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_039a040_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_039a040_gsd5_s2048/images,0,0 +47,2023,KHQP,042a045,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_042a045_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_042a045_gsd5_s2048/images,0,0 +48,2023,KHQP,046a049,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_046a049_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_046a049_gsd5_s2048/images,0,0 +49,2023,KHQP,050a058,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_050a058_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_050a058_gsd5_s2048/images,0,0 +50,2023,KHQP,053a059a060,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_053a059a060_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_053a059a060_gsd5_s2048/images,0,0 +51,2023,KHQP,054a056,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_054a056_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_054a056_gsd5_s2048/images,0,0 +52,2023,KHQP,063a143,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_063a143_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_063a143_gsd5_s2048/images,0,0 +53,2023,KHQP,073a074a080,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_073a074a080_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_073a074a080_gsd5_s2048/images,0,0 +54,2023,KHQP,072a081a082,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_072a081a082_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_072a081a082_gsd5_s2048/images,0,0 +55,2023,KHQP,076,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_076_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_076_gsd5_s2048/images,0,0 +56,2023,KHQP,077a078,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_077a078_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_077a078_gsd5_s2048/images,0,0 +57,2023,KHQP,091a097a102a104,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_091a097a102a104_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_091a097a102a104_gsd5_s2048/images,0,0 +58,2023,KHQP,087a088a089a090,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_087a088a089a090_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_087a088a089a090_gsd5_s2048/images,0,0 +59,2023,KHQP,094,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_094_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_094_gsd5_s2048/images,0,0 +60,2023,KHQP,105a0111,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_105a0111_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_105a0111_gsd5_s2048/images,0,0 +61,2023,KHQP,112a113,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_112a113_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_112a113_gsd5_s2048/images,0,0 +62,2023,KHQP,114a116a118,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_114a116a118_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_114a116a118_gsd5_s2048/images,0,0 +63,2023,KHQP,115,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_115_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_115_gsd5_s2048/images,0,0 +64,2023,KHQP,117,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_117_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_117_gsd5_s2048/images,0,0 +65,2023,KHQP,119a120,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_119a120_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_119a120_gsd5_s2048/images,0,0 +66,2023,KHQP,034a131,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_034a131_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_034a131_gsd5_s2048/images,0,0 +67,2023,KHQP,132a133,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_132a133_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_132a133_gsd5_s2048/images,0,0 +68,2023,KHQP,057,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_057_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_057_gsd5_s2048/images,0,0 +69,2023,KHQP,066a142,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_066a142_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_066a142_gsd5_s2048/images,0,0 +70,2023,KHQP,106a107a138a139,S3,1,0,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_106a107a138a139_gsd5_s2048/annfiles,/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_106a107a138a139_gsd5_s2048/images,0,0 diff --git a/data.db b/data.db index 8825f05a8481c556842ec05c392785db7fdb6357..1f8f0ca41794ff679112df1f227003de34e7ed78 100644 GIT binary patch delta 41 xcmZp8z}WDBae_3X=|mZ4M$?T6)1(*F+g-My`zs)1(4lDow diff --git a/main.py b/main.py index 9358efc..e50069e 100644 --- a/main.py +++ b/main.py @@ -6,12 +6,18 @@ from loguru import logger from bak.init_data import BatchDataRead from db_utils import BatchData, session +PAGE_TITLE = "training data configer" +st.set_page_config( + page_title=PAGE_TITLE, + page_icon="🧊", + initial_sidebar_state="expanded", +) + def get_data_from_db(): logger.debug("init") db_objs = session.query(BatchData).all() - res = (BatchDataRead.from_orm(db_obj).dict() for db_obj in db_objs) - return list(res) + return [BatchDataRead.from_orm(db_obj).dict() for db_obj in db_objs] st.session_state.setdefault('data_table', get_data_from_db()) @@ -19,6 +25,10 @@ st.session_state.setdefault('data_table', get_data_from_db()) df = pd.DataFrame(data=st.session_state.data_table) +def train(): + ... + + def update_handler(): edited_rows = st.session_state['edited_info'].get('edited_rows') for id_, update_data in edited_rows.items(): @@ -29,25 +39,25 @@ def update_handler(): session.commit() -edited_df = st.data_editor(df, key="edited_info", - hide_index=True, - use_container_width=True, - on_change=update_handler, - column_order=('id', 'year', 'census_batch', 'id_code', 'precision', 'is_train', 'is_validation'), - column_config={ - "year": st.column_config.NumberColumn("年份", format="%d 年", ), - 'census_batch': "普查批次", - 'id_code': "编号", - 'precision': "精度", - 'is_train': "是否是训练集", - 'is_validation': "是否是验证集", - }) +data_frame_container = st.container() +with data_frame_container: + edited_df = st.data_editor( + df, key="edited_info", + hide_index=True, + use_container_width=True, + on_change=update_handler, + column_order=('id', 'year', 'census_batch', 'id_code', 'precision', 'is_train', 'is_validation'), + column_config={ + "year": st.column_config.NumberColumn("年份", format="%d 年", ), + 'census_batch': "普查批次", + 'id_code': "编号", + 'precision': "精度", + 'is_train': "是否是训练集", + 'is_validation': "是否是验证集", + }) -# favorite_command = edited_df.loc[edited_df["id"].idxmax()] -# data_table_change_info = st.session_state['edited_info'] -# logger.debug(f"{data_table_change_info=}") - -# st.rerun() -# st.session_state.data_table = get_data_from_db() - -# st.rerun() +with st.sidebar: + st.divider() + st.text_input("username") + st.divider() + st.button("启动", on_click=train)