A device for image classification comprising a convolutional neural network configured to generate a plurality of probability values, each probability value being linked to a respective one of a plurality of predetermined classes and indicating the probability that the image or a pixel of the image is associated with the respective class, and the convolutional neural network comprises a plurality of convolutional blocks and each of the convolutional blocks comprises: a first convolutional layer configured to perform a pointwise convolution using a first kernel, a second convolutional layer configured to perform a depthwise convolution using a second kernel, wherein the second kernel has one of a single row and a single column, a third convolutional layer configured to perform a depthwise convolution using a third kernel, wherein the third kernel has a single column if the second kernel has a single row, and the third kernel has a single row if the second kernel has a single column, and a fourth convolutional layer configured to perform a convolution using a fourth kernel.