How can we find dominant color of an image in pytorch from the incoming Tensor.have to remove all the white pixels outside the edges of the tensor ?.
The above operations needs to take place in the training loop …
I have built this using PIL in my above training loop