test.py

#!/usr/bin/python3

import argparse
import itertools
import os
import torchvision.transforms as transforms
from torch.utils.data import DataLoader
from torch.autograd import Variable
from PIL import Image
import torch
import torch.nn.functional as F
import numpy as np
from trainer import P2p_Trainer_v6
import yaml

def get_config(config):
    with open(config, 'r') as stream:
        return yaml.load(stream)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--config', type=str, default='./Yaml/mrgan.yaml', help='Path to the config file.')
    opts = parser.parse_args()
    config = get_config(opts.config)
    print("======config=======")
    print(config)

    trainer = P2p_Trainer_v6(config)

    trainer.test()
    
if __name__ == '__main__':
    main()