search

jiwoon-ahn / irn

Weakly Supervised Learning of Instance Segmentation with Inter-pixel Relations, CVPR 2019 (Oral)
MIT License
519 stars 100 forks source link

Fixes on run.py arguemtns #47

Open wow056 opened 1 year ago

wow056 commented 1 year ago

Includes

  1. Fix sample_run.py step arguments to store boolean value. (Previously, it stored string value like 'True' if I specified arguements)
  2. Grouping arguments to make --help readable 
    
> python run_sample.py -h
usage: run_sample.py [-h] --voc12_root VOC12_ROOT [--num_workers NUM_WORKERS]
                 [--train_list TRAIN_LIST] [--val_list VAL_LIST]
                 [--infer_list INFER_LIST]
                 [--chainer_eval_set CHAINER_EVAL_SET]
                 [--cam_network CAM_NETWORK]
                 [--cam_crop_size CAM_CROP_SIZE]
                 [--cam_batch_size CAM_BATCH_SIZE]
                 [--cam_num_epoches CAM_NUM_EPOCHES]
                 [--cam_learning_rate CAM_LEARNING_RATE]
                 [--cam_weight_decay CAM_WEIGHT_DECAY]
                 [--cam_eval_thres CAM_EVAL_THRES]
                 [--cam_scales CAM_SCALES] [--conf_fg_thres CONF_FG_THRES]
                 [--conf_bg_thres CONF_BG_THRES]
                 [--irn_network IRN_NETWORK]
                 [--irn_crop_size IRN_CROP_SIZE]
                 [--irn_batch_size IRN_BATCH_SIZE]
                 [--irn_num_epoches IRN_NUM_EPOCHES]
                 [--irn_learning_rate IRN_LEARNING_RATE]
                 [--irn_weight_decay IRN_WEIGHT_DECAY] [--beta BETA]
                 [--exp_times EXP_TIMES]
                 [--ins_seg_bg_thres INS_SEG_BG_THRES]
                 [--sem_seg_bg_thres SEM_SEG_BG_THRES]
                 [--log_name LOG_NAME]
                 [--cam_weights_name CAM_WEIGHTS_NAME]
                 [--irn_weights_name IRN_WEIGHTS_NAME]
                 [--cam_out_dir CAM_OUT_DIR]
                 [--ir_label_out_dir IR_LABEL_OUT_DIR]
                 [--sem_seg_out_dir SEM_SEG_OUT_DIR]
                 [--ins_seg_out_dir INS_SEG_OUT_DIR] [--train_cam]
                 [--make_cam] [--eval_cam] [--cam_to_ir_label]
                 [--train_irn] [--make_ins_seg] [--eval_ins_seg]
                 [--make_sem_seg] [--eval_sem_seg]

optional arguments: -h, --help show this help message and exit

Environment: --voc12_root VOC12_ROOT Path to VOC 2012 Devkit, must contain ./JPEGImages as subdirectory. --num_workers NUM_WORKERS

Dataset: --train_list TRAIN_LIST --val_list VAL_LIST --infer_list INFER_LIST voc12/train_aug.txt to train a fully supervised model, voc12/train.txt or voc12/val.txt to quickly check the quality of the labels. --chainer_eval_set CHAINER_EVAL_SET

Class Activation Map: --cam_network CAM_NETWORK --cam_crop_size CAM_CROP_SIZE --cam_batch_size CAM_BATCH_SIZE --cam_num_epoches CAM_NUM_EPOCHES --cam_learning_rate CAM_LEARNING_RATE --cam_weight_decay CAM_WEIGHT_DECAY --cam_eval_thres CAM_EVAL_THRES --cam_scales CAM_SCALES Multi-scale inferences

Mining Inter-pixel Relations: --conf_fg_thres CONF_FG_THRES --conf_bg_thres CONF_BG_THRES

Inter-pixel Relation Network (IRNet): --irn_network IRN_NETWORK --irn_crop_size IRN_CROP_SIZE --irn_batch_size IRN_BATCH_SIZE --irn_num_epoches IRN_NUM_EPOCHES --irn_learning_rate IRN_LEARNING_RATE --irn_weight_decay IRN_WEIGHT_DECAY

Random Walk Params: --beta BETA --exp_times EXP_TIMES Hyper-parameter that controls the number of random walk iterations,The random walk is performed 2^{exp_times}. --ins_seg_bg_thres INS_SEG_BG_THRES --sem_seg_bg_thres SEM_SEG_BG_THRES

Output Path: --log_name LOG_NAME --cam_weights_name CAM_WEIGHTS_NAME --irn_weights_name IRN_WEIGHTS_NAME --cam_out_dir CAM_OUT_DIR --ir_label_out_dir IR_LABEL_OUT_DIR --sem_seg_out_dir SEM_SEG_OUT_DIR --ins_seg_out_dir INS_SEG_OUT_DIR

Step: --train_cam --make_cam --eval_cam --cam_to_ir_label --train_irn --make_ins_seg --eval_ins_seg --make_sem_seg --eval_sem_seg