Rather than get a single classification for an entire image we would like a classification per pixel.
We have 6 classes:
Spheroid, Disk, Irregular, Point Source, Unknown, Background
The model: