Hi,
DeepCluster expects the input to be a RGB image so that channel should be 3 instead of 1. If you look at the implementation of AlexNet in that repo, in the forward method, they are applying sobel filter that converts the RGB image to grayscale and then applies sobel filter.
Since in your case, you’re passing single channel image directly, so that portion of code will throw shape error. To fix it, you could either send RGB images or remove the grayscale portion of the sobel filter code.


Image Source: A Visual Exploration of DeepCluster 2