def main():
logging.basicConfig(level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(message)s",
datefmt="%Y-%m-%d %H:%M:%S",
filename="log.txt",
filemode='w')
console = logging.StreamHandler()
console.setLevel(logging.INFO)
formatter = logging.Formatter('%(message)s')
console.setFormatter(formatter)
logging.getLogger('').addHandler(console)
model = FrontNet(PreActBlock, [1, 1, 1])
ModelManager.Read('../PyTorch/Models/Dronet-032.pt', model)
DATA_PATH = "/Users/usi/PycharmProjects/data/"
[x_test,
y_test] = DataProcessor.ProcessTestDataGray(DATA_PATH + "patterns.pickle",
60, 108)
#x_test = x_test[:500]
#y_test = y_test[:500]
test_set = Dataset(x_test, y_test)
params = {'batch_size': 1, 'shuffle': False, 'num_workers': 0}
test_generator = data.DataLoader(test_set, **params)
trainer = ModelTrainer(model)
valid_loss_x, valid_loss_y, valid_loss_z, valid_loss_phi, outputs, gt_labels = trainer.ValidateSingleEpoch(
test_generator)
Viz4PoseVariables(x_test, y_test, outputs)
