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)
