parser.add_argument("--train_irn_pass", default=True) parser.add_argument("--make_ins_seg_pass", default=True) parser.add_argument("--eval_ins_seg_pass", default=True) parser.add_argument("--make_sem_seg_pass", default=True) parser.add_argument("--eval_sem_seg_pass", default=True) args = parser.parse_args() os.makedirs("sess", exist_ok=True) os.makedirs(args.cam_out_dir, exist_ok=True) os.makedirs(args.ir_label_out_dir, exist_ok=True) os.makedirs(args.sem_seg_out_dir, exist_ok=True) os.makedirs(args.ins_seg_out_dir, exist_ok=True) pyutils.Logger(args.log_name + '.log') print(vars(args)) if args.train_cam_pass is True: print("[INFO] Train CAM") import step.train_cam timer = pyutils.Timer('step.train_cam:') step.train_cam.run(args) if args.make_cam_pass is True: print("[INFO] Make CAM") import step.make_cam timer = pyutils.Timer('step.make_cam:') step.make_cam.run(args)
parser.add_argument("--cam_to_ir_label_pass", default=True) parser.add_argument("--train_irn_pass", default=True) parser.add_argument("--make_ins_seg_pass", default=True) parser.add_argument("--eval_ins_seg_pass", default=True) parser.add_argument("--make_sem_seg_pass", default=True) parser.add_argument("--eval_sem_seg_pass", default=True) args = parser.parse_args() os.makedirs("sess", exist_ok=True) os.makedirs(args.cam_out_dir, exist_ok=True) os.makedirs(args.ir_label_out_dir, exist_ok=True) os.makedirs(args.sem_seg_out_dir, exist_ok=True) os.makedirs(args.ins_seg_out_dir, exist_ok=True) pyutils.Logger(args.log_name + ".log") print(vars(args)) if args.train_cam_pass is True: import step.train_cam timer = pyutils.Timer("step.train_cam:") step.train_cam.run(args) if args.make_cam_pass is True: import step.make_cam timer = pyutils.Timer("step.make_cam:") step.make_cam.run(args) if args.eval_cam_pass is True: