Training CNN on training set with different numbers of images in the training set

3 次查看(过去 30 天)
Hi, I'm doing transfer learning to train squeezenet on a database of microcopy images. I have two classes that have different numbers of images (class 1: 200, class2: 400). Is there a way in Matlab to assign lower weight to the images in class 2, so that the classification is not biased towards the most numerous class? Thank you

采纳的回答

Srivardhan Gadila
Srivardhan Gadila 2021-3-23
You can enable ClassWeights for classificationLayer or you can also use focalLossLayer.

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

产品


版本

R2020b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by