遥感影像场景分类预测
最后更新 2020/03/25 23:02
阅读 13781
PyTorch SENet
heshuting555
76
获得赞0
发布的文章1
答辩的项目Batch大小为30,循环次数为10次,通过在线上环境完成训练,模型最优精度评分为92.55。
最后更新 2020/03/25 23:02
阅读 13781
PyTorch SENet
本赛题共有45个类别,涉及多分类问题。必定会存在样本不平衡问题。
核心思路
数据预处理:每个类别的样本个数不一样,故采用Imbalanced Dataset Sampler调整每个类别的权重最后使得整个样本群每个类别平衡。
以下是 Label Smoothing 的代码:
class ImbalancedDatasetSampler(torch.utils.data.sampler.Sampler):
ㅤdef __init__(self, dataset, indices=None, num_samples=None):
ㅤㅤㅤself.indices = list(range(len(dataset)))
ㅤㅤㅤㅤㅤif indices is None else indices
ㅤㅤㅤself.num_samples = len(self.indices)
ㅤㅤㅤㅤㅤif num_samples is None else num_samples
# distribution of classes in the dataset
ㅤㅤㅤlabel_to_count = {}
请先绑定您的微信账号 点击立即绑定
敬请谅解,如有疑问请联系FlyAI客服