sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13:01 +00:00
Code Issues Projects Releases Packages 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 # Loading the Colormap
colormap = loadmat( colormap = loadmat(
"./instance-level_human_parsing/instance-level_human_parsing/human_colormap.mat" os.path.join(DIR_OUTPUT, "human_colormap.mat")
)["colormap"] )["colormap"]
colormap = colormap * 100 colormap = colormap * 100
colormap = colormap.astype(np.uint8) colormap = colormap.astype(np.uint8)