Closed sarmientoj24 closed 2 years ago
Since it produces feature maps, how do I "flatten" the feature maps for the classifier?
Hi @sarmientoj24, Sorry for my late response. Have a look at the image classification code here. This class just wraps the model for classification.
Since it produces feature maps, how do I "flatten" the feature maps for the classifier?