def pdc_train(dataset_config, train_config, dataset_name, logging_dir, num_iterations, dimension):
# print("training args")
# print(dataset_config)
# print(train_config)
# print(dataset_name)
# print(logging_dir)
# print(num_iterations)
# print(dimension)
print('dataset_name')
print(dataset_name)
dataset = SpartanDataset(config=dataset_config)
d = dimension # the descriptor dimension
name = dataset_name.split('/')[-1] + "_%d" %(d)
train_config["training"]["logging_dir_name"] = name
print('logging dir name')
print(name)
train_config["training"]["logging_dir"] = logging_dir
train_config["dense_correspondence_network"]["descriptor_dimension"] = d
train_config["training"]["num_iterations"] = num_iterations
print "training descriptor of dimension %d" %(d)
start_time = time.time()
train = DenseCorrespondenceTraining(dataset=dataset, config=train_config)
train.run()
end_time = time.time()
print "finished training descriptor of dimension %d using time %.2f seconds" %(d, end_time-start_time)
def train(self):
# This should take about ~12-15 minutes with a GTX 1080 Ti
# All of the saved data for this network will be located in the
# code/data_volume/pdc/trained_models/tutorials/caterpillar_3 folder
descr_dim = self.train_config["dense_correspondence_network"][
"descriptor_dimension"]
print("training descriptor of dimension %d" % (descr_dim))
train = DenseCorrespondenceTraining(dataset=self.dataset,
config=self.train_config)
train.run()
print("finished training descriptor of dimension %d" % (descr_dim))
from dense_correspondence.dataset.spartan_dataset_masked import SpartanDataset
logging.basicConfig(level=logging.INFO)
from dense_correspondence.evaluation.evaluation import DenseCorrespondenceEvaluation
config_filename = os.path.join(utils.getDenseCorrespondenceSourceDir(), 'config', 'dense_correspondence',
'dataset', 'composite', 'toy.yaml')
config = utils.getDictFromYamlFilename(config_filename)
train_config_file = os.path.join(utils.getDenseCorrespondenceSourceDir(), 'config', 'dense_correspondence',
'training', 'toy_training.yaml')
train_config = utils.getDictFromYamlFilename(train_config_file)
dataset = SpartanDataset(config=config)
logging_dir = "/home/zhouxian/git/pytorch-dense-correspondence/pdc/trained_models/tutorials"
d = 3 # the descriptor dimension
name = "toy_hacker_%d" %(d)
train_config["training"]["logging_dir_name"] = name
train_config["training"]["logging_dir"] = logging_dir
train_config["dense_correspondence_network"]["descriptor_dimension"] = d
TRAIN = True
EVALUATE = True
if TRAIN:
print "training descriptor of dimension %d" %(d)
train = DenseCorrespondenceTraining(dataset=dataset, config=train_config)
train.run()
# train.run_from_pretrained('/home/zhouxian/git/pytorch-dense-correspondence/pdc/trained_models/tutorials/backup/toy_hack_3')
print "finished training descriptor of dimension %d" %(d)