def load(self, conf_file):
conf = ConfigLoader(conf_file)
result_tcp = conf.tcp_configloder()
result_redis = conf.redis_configloder()
if result_tcp is False or result_redis is False:
logging.critical('Failed configloder...')
return False
self.conf = conf
return True
import os
from databuilder.rvl_cdip import build_dataset
from utils.configloader import ConfigLoader
from models.srcnn import Model
# from models.srresnet import Model
DATA_DIR = '/home/super-videt/Storage/Resource/CV/dataset'
NEW_DIR = './datasets'
if __name__ == '__main__':
# os.environ["CUDA_VISIBLE_DEVICES"] = "0"
srcnn_path = "results/checkpoints/srcnn/srcnn_param_batch4_lr0.001_epoch10.pkl"
cfg = ConfigLoader('configs/srcnn.cfg', 'default')
cfg = ConfigLoader('configs/srcnn.cfg', 'rvl_cdip')
model = Model(cfg)
model.train(srcnn_path=srcnn_path,
random_scale=False,
rotate=False,
fliplr=False,
fliptb=False)
model.test(srcnn_path=srcnn_path)
# os.environ["CUDA_VISIBLE_DEVICES"] = "1"
# srcnn_path = "results/checkpoints/srresnet/srresnet_param_batch4_lr0.001_epoch3.pkl"
# cfg = ConfigLoader('configs/srresnet.cfg', 'default')
# cfg = ConfigLoader('configs/srresnet.cfg', 'rvl_cdip')
# model = Model(cfg)
# model.train(srcnn_path=srcnn_path,
class ConfigError(Exception):
pass
if __name__ == "__main__":
try:
makelog('topologyMysql')
logging.info('START---------------')
root_path = os.path.dirname(os.path.abspath(__file__))
config_file = os.path.join(root_path, 'configs', 'topology.ini')
conf = ConfigLoader(config_file)
result_redis = conf.redis_configloder()
result_mysql = conf.mysql_configloder()
if result_redis is False or result_mysql is False:
raise ConfigError('Failed configloader...')
mysql_client = connect_mysql(conf)
if mysql_client:
cur = mysql_client.cursor(DictCursor)
connect_redis = connect_redis(conf)
if connect_redis:
while True:
try:
data = pop_redis(connect_redis, conf)
if __name__ == '__main__':
# build_dataset(data_dir=DATA_DIR, new_dir=NEW_DIR, mode='test')
# os.environ["CUDA_VISIBLE_DEVICES"] = "1"
edgenet_path = "edgenet_param_batch4_lr0.002_epoch100.pkl"
srcnn_path = "results/checkpoints/srcnn/srcnn_param_batch4_lr0.001_epoch3.pkl"
sr2x1_path = "results/checkpoints/srunitnet/srnet2x1_param_batch4_lr0.0005_epoch1.pkl"
sr2x2_path = "results/checkpoints/srunitnet/srnet2x2_param_batch4_lr0.0005_epoch1.pkl"
srresnet_path = "results/checkpoints/srresnet/srresnet_param_batch4_lr0.001_epoch3.pkl"
sr2x1_edge_path = "results/checkpoints/srunitnet/srnet2x1_param_batch4_lr0.0005_epoch47.pkl"
sr2x2_edge_path = "results/checkpoints/srunitnet/srnet2x2_param_batch4_lr0.0005_epoch47.pkl"
# sr2x1_path = None
# sr2x2_path = None
cfg = ConfigLoader('configs/default.cfg', 'document')
cfg = ConfigLoader('configs/default.cfg', 'rvl_cdip')
# cfg = ConfigLoader('configs/default.cfg', 'test')
model = Model(cfg)
model.train(sr2x1_path=sr2x1_path, sr2x2_path=sr2x2_path, edgenetpath=edgenet_path, srcnn_path=srcnn_path, srresnet_path=srresnet_path,
random_scale=False, is_fine_tune=False, rotate=False, fliplr=False, fliptb=False)
# model.train(sr2x1_path=sr2x1_path, sr2x2_path=sr2x2_path, edgenetpath=edgenet_path,
# random_scale=False, is_fine_tune=True, rotate=False, fliplr=False, fliptb=False)
# model.train(sr2x1_path=sr2x1_path, sr2x2_path=sr2x2_path, edgenetpath=edgenet_path,
# random_scale=False, is_fine_tune=True, rotate=False, fliplr=False, fliptb=False)
# model.test(sr2x1_path=sr2x1_path, sr2x2_path=sr2x2_path)
# model.test(srcnn_path=srcnn_path)
# sr2x1_edge_path = "results/checkpoints/srunitnet/srnet2x1_param_batch4_lr0.0005_epoch47.pkl"
# sr2x2_edge_path = "results/checkpoints/srunitnet/srnet2x2_param_batch4_lr0.0005_epoch47.pkl"
# sr2x1_none_path = "results/checkpoints/srunitnet/srnet2x1_param_batch4_lr0.001_epoch25.pkl"
