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()