https://github.com/IGLICT/PRS-Net_stamp
Raw File
Tip revision: 42710b64de733f1c45aeb9b011585f159cf3ac75 authored by Lingxiao Zhang on 22 February 2021, 04:22:16 UTC
Update README.MD
Tip revision: 42710b6
models.py

def create_PRSNet(opt):
    from .PRSNet import PRSNet, Inference
    if opt.isTrain:
        model = PRSNet()
    else:
        model = Inference()
    model.initialize(opt)
    if opt.verbose:
        print("model [%s] was created" % (model.name()))
    return model
back to top