search

kbressem / faimed3d

Extension to fastai for volumetric medical data
MIT License
32 stars 14 forks source link

implemented 4D data support for CNN #21

Closed kbressem closed 3 years ago

kbressem commented 3 years ago

Proposed solution for #13 for CNN.

Adds a new Wrapper class for CNN-body if n_inp is > 1. Adds 1x1 convolutional layer to model head to pool features from multiple inputs.