mapbox / robosat

Semantic segmentation on aerial and satellite imagery. Extracts features such as: buildings, parking lots, roads, water, clouds
MIT License
2.02k stars 382 forks source link

No output from predict #128

Closed wboykinm closed 5 years ago

wboykinm commented 5 years ago

Using this workflow - CPU-only, on a Macbook, I'm consistently producing no outputs here:

root@f5a70d76856f:/app# ./rs predict --batch_size 1 --checkpoint rutland/tmp/checkpoint-00001-of-00001.pth --tile_size 256 --model config/model-unet-rutland.toml --dataset config/dataset-building.toml rutland probs
Eval: 0batch [00:00, ?batch/s]

The 0batch eval bit seems like a pretty clear indicator that nothing's happening, but the question in my mind is "Just how fundamental of a config screwup is this on my part?" The train step produced some 436MB checkpoints, but the associated "history" chart looks . . . well . . . mostly dead:

history-00002-of-00002

Is this lack of output likely the result of

(🙏 for the awesome toolset and docs; mostly more helpful to the semiliterate like me than I have any right to expect)

ocourtin commented 5 years ago

Directly related to your question:

With predict command, the tile directory to pass is the one containing tiles images, so for instance rather than rutland you should use: rutland/validation/images
(or any other predict dedicated tiles directory)

Indirectly related to your question:

Enjoy :)

wboykinm commented 5 years ago

perfect

wboykinm commented 5 years ago

I've made the recommended adjustments and am running this again. As each training epoch takes ~1hr on this machine, it'll be a little while before I post results 😄

Thanks again!

wboykinm commented 5 years ago

ZING:

47859

Thanks, @ocourtin!