import os import frcnn_utils import init_frcnn_utils from experiments.exp_utils import get_config_var, LoggerForSacred, Args from init_frcnn_utils import init_dataloaders_1s_1t, init_val_dataloaders_mt, init_val_dataloaders_1t, \ init_htcn_model_optimizer from sacred import Experiment ex = Experiment() from sacred.observers import MongoObserver enable_mongo_observer = False if enable_mongo_observer: vars = get_config_var() ex.observers.append( MongoObserver( url='mongodb://{}:{}@{}/admin?authMechanism=SCRAM-SHA-1'.format( vars["SACRED_USER"], vars["SACRED_PWD"], vars["SACRED_URL"]), db_name=vars["SACRED_DB"])) ex.captured_out_filter = lambda text: 'Output capturing turned off.' from dataclasses import dataclass import numpy as np import torch import torch.nn as nn from model.utils.config import cfg, cfg_from_file, cfg_from_list from model.utils.net_utils import adjust_learning_rate, save_checkpoint, FocalLoss, EFocalLoss
import os import frcnn_utils import distill_frcnn_utils from experiments.exp_utils import get_config_var, LoggerForSacred, Args import init_frcnn_utils from sacred import Experiment ex = Experiment() from sacred.observers import MongoObserver enable_mongo_observer = False if enable_mongo_observer: vars = get_config_var(is_eval=True) ex.observers.append(MongoObserver(url='mongodb://{}:{}@{}/admin?authMechanism=SCRAM-SHA-1'.format(vars["SACRED_USER"], vars["SACRED_PWD"], vars["SACRED_URL"]), db_name=vars["SACRED_DB"])) ex.captured_out_filter = lambda text: 'Output capturing turned off.' from dataclasses import dataclass import numpy as np import torch import torch.nn as nn from model.utils.config import cfg, cfg_from_file, cfg_from_list from model.utils.net_utils import adjust_learning_rate, save_checkpoint, FocalLoss, EFocalLoss from model.faster_rcnn.resnet import resnet as n_resnet