File size: 693 Bytes
be2715b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from segmentation_2d.train_R50_seg_adam_optimizer_2d import train_2d_R50
from segmentation_3d.train_R50_seg_adam_optimizer_3d import train_3d_R50
from utils.func import (
parse_config,
load_config
)
if __name__=="__main__":
yml_args = parse_config()
cfg = load_config(yml_args.config)
assert cfg.base.is_2D + cfg.base.is_3D == 1
if cfg.base.is_2D:
if cfg.base.is_R50:
train_2d_R50(yml_args, cfg)
if cfg.base.is_SAMVIT:
train_2d_SAMVIT(yml_args, cfg)
if cfg.base.is_3D:
if cfg.base.is_R50:
train_3d_R50(yml_args, cfg)
if cfg.base.is_SAMVIT:
train_3d_SAMVIT(yml_args, cfg) |