I’m developing a U_Net Model for Segmentation. My Train dataset shape is
[{‘image’, [32, 128, 128, 2]};
{‘detection’, [32, 128, 128, 2]}]
Such that { ‘image’, [number of MRI slices, height, width, len([‘T2W’, ‘ADC’])] }
{ ‘detection’, [number of MRI slices, height, width, number of classes ] }.