Marvee Amasi
DIIDevHeads IoT Integration Server
•Created by wafa_ath on 7/19/2024 in #middleware-and-os
Issue with 4-Dimensional Input Error in PyTorch Model Inference
Hi @wafa_ath see what's happening here ur images prolly is coming in as single tensors of size (224, 224, 3) for RGB channels, so the model expects an extra dimension at the beginning to represent the batch size. Reshape your image or instead add a batch dimension of size 1
6 replies