def startTest(self,params):
params['signalInfo'].emit(0, "开始检测...")
dataset = params['modelName'] # 'cus_mnist_2'
dataroot = params['path'] # 'E:\ProjectSet\Pycharm\WAIBAO\cus_mnist2'
opt = Options().parse(dataset)
opt.isTrain = False
opt.load_weights = True
opt.signal = params['signal']
opt.signalInfo = params['signalInfo']
opt.lr = self.modelsData[dataset]['opt']['lr']
opt.nz = self.modelsData[dataset]['opt']['nz']
opt.batchsize = self.modelsData[dataset]['opt']['batchsize']
opt.dataroot = dataroot
print(opt)
self.modelTest = MyTest(opt, [self.modelsData[dataset]['minVal'], self.modelsData[dataset]['maxVal'], self.modelsData[dataset]['proline']])
self.modelTest.start()
def startTrain(self, params):
params['signalInfo'].emit(0, "开始训练...")
dataset = params['name'] #'cus_mnist_2'
dataroot = params['path'] #'E:\ProjectSet\Pycharm\WAIBAO\cus_mnist2'
opt = Options().parse(dataset)
opt.signal = params['signal']
opt.load_weights = False
opt.signalInfo = params['signalInfo']
opt.lr = params['-lr']
opt.batchsize = params['-batchsize']
opt.niter = params['-niter']
opt.nz = params['-nz']
opt.desc = params['info']
opt.dataroot = dataroot
# opt.isize = 128
print(opt)
self.modelTrain = MyModel(opt)
self.modelTrain.start()
