[jpace121]

I’ve worked on plenty of problems where all you need is to find what pixels are a specific color. While I’m sure a NN could be setup for that, why make things harder than they have to be?

[albertzeyer]

Why harder? My point is, by using some DL framework, you would make this simpler. You don't need to have a big NN for that, or even any NN at all, and could still use some DL framework, and I would assume this is still easier and probably faster. E.g. finding what pixels are a specific color, this would be sth like:

    torch.all(image == torch.tensor([255,0,0]), dim=-1).nonzero()

To get all pixel pairs (x,y) where the color is red (255,0,0) of the image of shape (width,height,channels).

[matsemann]

How do you generate the torch vector to do this? Often it's OpenCV!