search

warmspringwinds / pytorch-segmentation-detection

Image Segmentation and Object Detection in Pytorch
MIT License
745 stars 170 forks source link

Convert map outputs to lists for Python3 support #18

Closed erasaur closed 6 years ago

erasaur commented 6 years ago

Hey Daniil, thanks for your work on this repo.

I noticed that some places in the code use the output of the map function directly, which is a problem for Python3 since the function returns a map object instead of a list. I've just gone ahead and converted instances of map object to list in a bunch of places -- I noticed you did this for some files already, but it's currently breaking the cityscapes dataset, which I'm currently trying to use :-)

The Python2 docs guarantee the result of map is always a list so I assume this shouldn't break anything.

warmspringwinds commented 6 years ago

Hey @erasaur -- great job, thank you so much!