config.py

import os
import yaml
import easydict
from os.path import join


class Dataset:
    def __init__(self, path, domains, files, prefix):
        self.path = path
        self.prefix = prefix
        self.domains = domains
        self.files = [(join(path, file)) for file in files]
        self.prefixes = [self.prefix] * len(self.domains)


import argparse
parser = argparse.ArgumentParser(description='Code for *Learning to Transfer Examples for Partial Domain Adaptation*',
                                 formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--config', type=str, default='config.yaml', help='/path/to/config/file')

args = parser.parse_args()

config_file = args.config

args = yaml.load(open(config_file))

save_config = yaml.load(open(config_file))

args = easydict.EasyDict(args)

dataset = None
if args.data.dataset.name == 'office':
    dataset = Dataset(
    path=args.data.dataset.root_path,
    domains=['amazon', 'dslr', 'webcam'],
    files=[
        'amazon_31_list.txt',
        'dslr_31_list.txt',
        'webcam_31_list.txt'
    ],
    prefix=args.data.dataset.root_path)
elif args.data.dataset.name == 'officehome':
    dataset = Dataset(
    path=args.data.dataset.root_path,
    domains=['Art', 'Clipart', 'Product', 'Real_World'],
    files=[
        'Art.txt',
        'Clipart.txt',
        'Product.txt',
        'Real_World.txt'
    ],
    prefix=args.data.dataset.root_path)
elif args.data.dataset.name == 'visda':
    dataset = Dataset(
    path=args.data.dataset.root_path,
    domains=['simulation', 'real'],
    files=[
        'simulation_image_list.txt',
        'real_image_list.txt'
    ],
    prefix=args.data.dataset.root_path)
elif args.data.dataset.name == 'C2I':
    dataset = Dataset(
    path=args.data.dataset.root_path,
    domains=['Caltech', 'ImageNet'],
    files=[
        'C2I_caltech256.txt',
        'C2I_imagenet84.txt'
    ],
    prefix=args.data.dataset.root_path)
elif args.data.dataset.name == 'I2C':
    dataset = Dataset(
    path=args.data.dataset.root_path,
    domains=['ImageNet', 'Caltech'],
    files=[
        'I2C_imagenet1000.txt',
        'I2C_caltech84.txt'
    ],
    prefix=args.data.dataset.root_path)
else:
    raise Exception(f'dataset {args.data.dataset.name} not supported!')
# print (args.data.dataset.source)
source_domain_name = dataset.domains[args.data.dataset.source]
target_domain_name = dataset.domains[args.data.dataset.target]
source_file = dataset.files[args.data.dataset.source]
target_file = dataset.files[args.data.dataset.target]
print ("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ")
print ("source_domain_name :", source_domain_name, "target_domain_name :", target_domain_name)