burke86 / astro_rcnn

Astro R-CNN: Instance Segmentation in Astronomical Images using Mask R-CNN Deep Learning
MIT License
43 stars 15 forks source link

Generalize code to allow training and inference with an arbitrary number of bands #5

Open burke86 opened 4 years ago

burke86 commented 4 years ago

Requires generalizing load_image_disk at the very least. May need to generalize Mask R-CNN code to handle > 3 image channels.