自己的数据"/>
如何修改pytorch官方实现的Densenet的分类数,用于跑自己的数据
pytorch官方实现的Densenet的分类数是1000类,当我们想用来跑自己的数据集时,类别可能没有这么多,这时就需要我们去更改网络最后的分类器
##修改分类器代码如下
model = models.densenet169(pretrained=True)
###重塑最后分类层 class_num为自己的类别数,可以直接填数字
model.classifier = torch.nn.Linear(model.classifier.in_features, class_num)
model = model.train()
更多推荐
如何修改pytorch官方实现的Densenet的分类数,用于跑自己的数据
发布评论