Определить количество входных нейронов в нейронной сети сегментации изображений с помощью fastai?

Question

Здравствуйте, я новичок в нейронных сетях, особенно в слоях и нейронах. Я делаю проект сегментации изображений, у меня есть 200 входных изображений и я правильно помечен. В fastai есть возможность использовать предварительно обученную модель для трансферного обучения. Я прошел обучение, использовал learn.summarry. Я не понимаю, как я могу определить количество нейронов во входном слое (исходные данные)? Я считаю, что это имеет какое-то отношение к форме, но я не уверен, как узнать количество входных нейронов во входном слое. Кто-нибудь может мне помочь, пожалуйста! Спасибо!

ImageDataBunch;

Train: LabelList (160 items)
x: SegmentationItemList
Image (3, 128, 128),Image (3, 128, 128),Image (3, 128, 128),Image (3, 128, 128),Image (3, 128, 128)
y: SegmentationLabelList
ImageSegment (1, 128, 128),ImageSegment (1, 128, 128),ImageSegment (1, 128, 128),ImageSegment (1, 128, 128),ImageSegment (1, 128, 128)
Path: /content/drive/My Drive/Umes2020/JPEGImages;

Valid: LabelList (40 items)
x: SegmentationItemList
Image (3, 128, 128),Image (3, 128, 128),Image (3, 128, 128),Image (3, 128, 128),Image (3, 128, 128)
y: SegmentationLabelList
ImageSegment (1, 128, 128),ImageSegment (1, 128, 128),ImageSegment (1, 128, 128),ImageSegment (1, 128, 128),ImageSegment (1, 128, 128)
Path: /content/drive/My Drive/Umes2020/JPEGImages;

Answer 1

Я предполагаю, что вы имели в виду количество каналов на входе, поскольку количество нейронов первого слоя не ограничено формой изображения.

Для обычного CNN количество каналов равно 3 ( RGB). Для полностью подключенной сети количество каналов составляет 3 * H * W, поскольку каждый пиксель в каждом цветовом канале (RGB) становится 1 сглаженным каналом.