DeepLabv3+: fix pathing again

This commit is contained in:
Starbeamrainbowlabs 2022-12-13 18:26:00 +00:00
parent 7e1f271bf4
commit 440e693dfc
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -210,7 +210,7 @@ plt.savefig(os.path.join(DIR_OUTPUT, "val_acc.png"))
# Loading the Colormap
colormap = loadmat(
os.path.join(DIR_OUTPUT, "human_colormap.mat")
os.path.join(os.path.dirname(DIR_OUTPUT), "human_colormap.mat")
)["colormap"]
colormap = colormap * 100
colormap = colormap.astype(np.uint8)