wangwen-whu / WTW-Dataset

This is an official implementation for the WTW Dataset in "Parsing Table Structures in the Wild " on table detection and table structure recognition.
Other
152 stars 15 forks source link

关于centernet的physical structure的指标 #17

Open ZHEGG opened 1 year ago

ZHEGG commented 1 year ago

在使用centernet的代码复现其physical structure中的P,R,F指标时,发现自己的结果为 1 在iou=0.9时检测结果为0 并可视化了一些结果,发现其大概是正常的 2 所以对您iou=0.9的指标有些疑惑 我使用了3090x4来复现您的结果其opt如下

==> torch version: 1.7.0+cu110 ==> cudnn version: 8004 ==> Cmd: ['src/main.py', 'ctdet', '--exp_id', 'wtw_lr_1024', '--dataset', 'wtw', '--batch_size', '32', '--master_batch', '9', '--lr', '1.25e-4', '--gpus', '0,1,2,3', '--num_epochs', '150', '--lr_step', '90,120', '--input_res', '1024'] ==> Opt: K: 100 aggr_weight: 0.0 agnostic_ex: False arch: dla_34 aug_ddd: 0.5 aug_rot: 0 batch_size: 32 cat_spec_wh: False center_thresh: 0.1 chunk_sizes: [9, 8, 8, 7] data_dir: /home/mdisk2/bianzhe/CenterNet/src/lib/../../data dataset: wtw debug: 0 debug_dir: /home/mdisk2/bianzhe/CenterNet/src/lib/../../exp/ctdet/wtw_lr_1024/debug debugger_theme: white demo: dense_hp: False dense_wh: False dep_weight: 1 dim_weight: 1 down_ratio: 4 eval_oracle_dep: False eval_oracle_hm: False eval_oracle_hmhp: False eval_oracle_hp_offset: False eval_oracle_kps: False eval_oracle_offset: False eval_oracle_wh: False exp_dir: /home/mdisk2/bianzhe/CenterNet/src/lib/../../exp/ctdet exp_id: wtw_lr_1024 fix_res: True flip: 0.5 flip_test: False gpus: [0, 1, 2, 3] gpus_str: 0,1,2,3 head_conv: 256 heads: {'hm': 1, 'wh': 2, 'reg': 2} hide_data_time: False hm_hp: True hm_hp_weight: 1 hm_weight: 1 hp_weight: 1 input_h: 1024 input_res: 1024 input_w: 1024 keep_res: False kitti_split: 3dop load_model: lr: 0.000125 lr_step: [90, 120] master_batch_size: 9 mean: [[[0.7733555 0.78804266 0.7961663 ]]] metric: loss mse_loss: False nms: False no_color_aug: False norm_wh: False not_cuda_benchmark: False not_hm_hp: False not_prefetch_test: False not_rand_crop: False not_reg_bbox: False not_reg_hp_offset: False not_reg_offset: False num_classes: 1 num_epochs: 150 num_iters: -1 num_stacks: 1 num_workers: 4 off_weight: 1 output_h: 256 output_res: 256 output_w: 256 pad: 31 peak_thresh: 0.2 print_iter: 0 rect_mask: False reg_bbox: True reg_hp_offset: True reg_loss: l1 reg_offset: True resume: False root_dir: /home/mdisk2/bianzhe/CenterNet/src/lib/../.. rot_weight: 1 rotate: 0 save_all: False save_dir: /home/mdisk2/bianzhe/CenterNet/src/lib/../../exp/ctdet/wtw_lr_1024 scale: 0.4 scores_thresh: 0.1 seed: 317 shift: 0.1 std: [[[0.28359458 0.26819503 0.26328853]]] task: ctdet test: False test_scales: [1.0] trainval: False val_intervals: 5 vis_thresh: 0.3 wh_weight: 0.1

期待您的回复!