Color mapping error

[jorisguerin]

Hi @YeLyuUT and thanks for your nice work.

I just ran your code downloading the checkpoint that is proposed in the readme.
Results look OK, but I believe the mapping from colors to pixel classes is wrong (does not represent the paper classes neither the mapping in UAVColorEncoder.py.

Quick fix is to replace the createColorTable from this file by the following:

def createColorTable(self):
    clr_tab = {}
    clr_tab['Building'] = [128, 0, 0]
    clr_tab['Static_Car'] = [192, 0, 192]
    clr_tab['Tree'] = [0, 128, 0]
    clr_tab['Clutter'] = [0, 0, 0]
    clr_tab['Moving_Car'] = [64, 0, 128]
    clr_tab['Road'] = [128, 64, 128]
    clr_tab['Human'] = [64, 64, 0]
    clr_tab['Vegetation'] = [128, 128, 0]
    return clr_tab

Cheers
@jorisguerin

[YeLyuUT]

Hi @jorisguerin, thanks for your question!

The colors and the objects are linked by the dictionary, the order in the dictionary should not bother the result.
I have noticed some different behaviors when using different OS, it should be fixed now.
If you still experience some problems, please let me know.

Happy new year!
@YeLyuUT