Cast target masks to int at creation and refactor loader

neptune-ai / open-solution-mapping-challenge

Open solution to the Mapping Challenge :earth_americas:

https://www.crowdai.org/challenges/mapping-challenge

MIT License

380 stars 96 forks source link

Cast target masks to int at creation and refactor loader #109

Open jakubczakon opened 6 years ago

jakubczakon commented 6 years ago

Distance loader does

Di = self.load_joblib(distance_filepath)
Di = Di.astype(np.uint8)
Si = self.load_joblib(size_filepath).astype(np.uint16)
Si = np.sqrt(Si).astype(np.uint16)

This should have been done at creation of Di and Si not here i believe.