https://github.com/IGLICT/PRS-Net_stamp
Tip revision: 42710b64de733f1c45aeb9b011585f159cf3ac75 authored by Lingxiao Zhang on 22 February 2021, 04:22:16 UTC
Update README.MD
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