def main(): f = sys.argv[1] r = int(getarg(2, 0)) inname = getarg(3, "data/r%d_lin.p" % (r)) print("Test with radius=%d and file %s and model %s" % (r, f, inname)) model = pickle.load(open(inname, "rb")) w, h = Image.open(f).size X = extract_x([f], r) dim = (h - 2 * r, w - 2 * r) y = model.predict(X) y = np.reshape(y, dim) y = denormalize(y) if np.any(np.isnan(y)): print("result of", f, "has nan values after denormalizing") img = Image.fromarray(y.astype(np.uint8)) outfold = str.replace(str.replace(inname, ".p", ""), "data/", "out/") outf = path.join(outfold, path.basename(f)) mkdir_p(outfold) img.save(outf)
def main(): inname = getarg(1,"data/train_r0.npz") outname = getarg(2,"data/r0_lin.p") modelname = getarg(3, "Linear") files = np.load(inname) X = files['X'] y = files['y'] model = eval(modelname)() model.fit(X,y) pickle.dump(model, open(outname, "wb"))
def main(): inname = getarg(1,"data/train_raw.npz") model_name = getarg(2, "Linear") model = eval(model_name)() print("Train Set: "+inname) print("Loading file") files = np.load(inname) X = files['X'] y = files['y'] print("Validating...") scores = model.cross_val_score(X,y) print("Model: "+model_name) print("MSE: %0.2f (+/- %0.2f)" % (scores.mean()*100, scores.std() * 200))
def main(): r = int(getarg(1, 0)) k = int(getarg(2, 0)) print("Extracting with radius=%d" % (r)) prefix = "data/train" if(k > 0): prefix += "_k"+str(k) train_files_x = find_files(prefix) X = extract_x(train_files_x,r) train_files_y = find_files("data/train_cleaned") y = extract_y(train_files_y,r) outname = prefix+"_r"+str(r)+".npz" np.savez(outname,X=X,y=y)
def main(): r = int(getarg(1, 0)) k = int(getarg(2, 3)) print("Extracting with radius=%d" % (r)) prefix = "data/train" outname = prefix+"_c_k"+str(k)+"_r"+str(r)+".npz" print("Saving to: "+ outname) train_files_x = find_files(prefix) X = extract_x(train_files_x,r) Xk = extract_x(find_files(prefix+"_k"+str(k)),r,True) X = np.concatenate([X,Xk], axis=1) train_files_y = find_files("data/train_cleaned") y = extract_y(train_files_y,r) np.savez(outname,X=X,y=y)