Use multiple workers in `rs predict`

mapbox / robosat

Semantic segmentation on aerial and satellite imagery. Extracts features such as: buildings, parking lots, roads, water, clouds

MIT License

2.02k stars 383 forks source link

Use multiple workers in `rs predict` #107

Closed daniel-j-h closed 6 years ago

daniel-j-h commented 6 years ago

Looks like we are not using multiple workers in rs predict. We should let the data loader use parallelism to help with buffering and loading files from disk.

https://github.com/mapbox/robosat/blob/d43d69feb7f5a4a21f3fe4ef05a0b51dc267911e/robosat/tools/predict.py#L34t

daniel-j-h commented 6 years ago

@ocourtin fixed this in https://github.com/mapbox/robosat/pull/112 - thanks! Training and prediction is faster now!