def get_ram_config():
config, unparsed = parser.parse_known_args()
ram_base = "ram"
config = NOD(**config.__dict__)
# config.data_dir = PROJECT_APP_PATH.user_data / ram_base / 'data'
config.data_dir = Path.home() / "Data" / "vision_sample_data"
config.ckpt_dir = PROJECT_APP_PATH.user_data / ram_base / "ckpt"
config.logs_dir = PROJECT_APP_PATH.user_log / ram_base / "logs"
config.plot_dir = PROJECT_APP_PATH.user_log / ram_base / "plots"
return config # , unparsed
from pathlib import Path from neodroidvision import PROJECT_APP_PATH from neodroidvision.detection import SSDBoxPredictor from neodroidvision.detection.single_stage.ssd.architecture.backbones import vgg_factory from warg import NOD base_cfg = NOD() base_cfg.data_dir = Path.home() / "Data" / "Vision" / "Detection" # base_cfg.DATA_DIR = Path.home() / "Data" / "Datasets" base_cfg.output_dir = PROJECT_APP_PATH.user_data / "results" # ---------------------------------------------------------------------------- # # Model # ---------------------------------------------------------------------------- # base_cfg.model = NOD() base_cfg.model.device = "cuda" # ---------------------------------------------------------------------------- # # Backbone # ---------------------------------------------------------------------------- # base_cfg.model.backbone = NOD() base_cfg.model.backbone.name = vgg_factory base_cfg.model.backbone.out_channels = (512, 1024, 512, 256, 256, 256) base_cfg.model.backbone.pretrained = True base_cfg.model.backbone.predictor_type = SSDBoxPredictor # ---------------------------------------------------------------------------- # # Head