err_mean = np.mean(depth_angMasked)
myplt.drawtextInAxesBttomLeft(
fig, ax, "mean:{:>5.2f} [mm]".format(err_mean), -0.008, 8)
x.append(zen_max), y.append(err_mean)
else:
err_mean = 0
myplt.drawtextInAxesBttomRight(fig, ax, "No area", -0.008, 8)
cnt += 1
if saveFlag:
myplt.my_savefig(
fig,
str(savefile.joinpath("{}.png".format(extractID(gtpath.name)))))
fig.show()
# scatter err by zenith angle
#---------------------------------
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(x, y)
for zenith_num in range(0, 90, zenith_step):
y_ex = np.array(y)[np.array(x) == zenith_num + zenith_step]
mn = np.mean(y_ex)
ax.text(zenith_num + zenith_step, 30, "{:.2f}".format(mn), ha="center")
# Calc error
ang_deg = calcAngError(gt, rec)
# consider mask area
if maskFlag:
maskpath = find_same_id(gtpath, maskpaths)
mask = cv2.imread(str(maskpath), -1).astype(np.float32)
ang_deg = extractMaskArea(ang_deg, mask, maskFlag)
noObj_mask = extractMaskArea(noObj_mask, mask, maskFlag)
s0_mask = extractMaskArea(s0_mask, mask, maskFlag)
ang_deg = ang_deg[noObj_mask * s0_mask]
mn, std = np.mean(ang_deg), np.std(ang_deg)
errs = np.append(errs, ang_deg)
err_perEpoch.append(np.mean(errs))
ax.plot(epochNums,
err_perEpoch,
label="{}_{}".format(label,
data_name.split("/")[1]),
color=color,
marker=shape)
ax.legend(bbox_to_anchor=(1.05, 1), loc="upper left")
if saveFlag:
myplt.my_savefig(fig, "{}.png".format(saveFlag))
f.close()
for line in lines:
line = my_split(line)
train_loss.append(float(line[1].split(":")[1]))
test_loss.append(float(line[2].split(":")[1]))
#----------------------
# plot
#----------------------
fig = plt.figure()
ax = fig.add_subplot(111)
x = range(1, len(train_loss) + 1)
ax.plot(x, test_loss, label="test")
ax.plot(x, train_loss, label="train")
ax.set_yscale("log")
ax.grid(which="both")
ax.set_title("loss on each epoch", fontsize=18)
#ax.set_ylim(10**0, 10**3)
ax.set_ylim(10**0, 10**4)
ax.set_xlabel("epoch number", fontsize=15)
ax.set_ylabel("value of loss", fontsize=15)
ax.legend(fontsize=12)
fig.show()
myplt.my_savefig(fig, path + "result.png")