Open nrweir opened 5 years ago
@jshermeyer I moved this over to the solaris repo, flagging you here so you know where it is.
I'm still open to putting this in solaris.raster
instead of part of the in-line augmentation pipeline in solaris.nets
if you'd rather do it that way. It would (almost definitely) be easier to do that way and create new image files rather than trying to do it in-line, would just require a little tweaking of the pipeline code to figure out how to pass those images into the model.
@nrweir commented on Wed Apr 17 2019
Potentially helpful link for making this work with GPU acceleration:
http://deeplearnphysics.org/Blog/2018-10-02-Pinning-data-to-GPU.html
Either way we'll probably want to have it able to work with both GPU and CPU versions. I'm not sure if there's currently a CPU/GPU flag in the config skeleton yaml file, but if not, we should add one.
@nrweir commented on Wed Apr 17 2019
Feel free to change any of the labels, etc. related to the issue if you want.