sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-06-30 18:44:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

deeplabv3+: fix colourmap

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-12-13 14:02:10 +00:00
parent 4d8ce792c9
commit 7e1f271bf4
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/deeplabv3_plus_test.py
View file

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