所在位置：
首页 >
营销资讯 >
行业动态 > classifier[6].out_features
classifier[6].out_features
时间：2022-03-16 16:15:01 | 来源：行业动态
时间：2022-03-16 16:15:01 来源：行业动态
是模型最终的输出数量，也就是分类数量。
- 因为AlexNet这个深度神经网络的鼻祖，是在2012年ILSVRC竞赛中以1000分类的ImageNet数据集作为测试标，一举拔得头筹而开创深度学习新时代，这个1000分类的AlexNet图像分类模型便成为这个领域的经典之作，因此在标准的AlexNet模型中就保留1000这个数字作为基准。
- 在这个避障应用中只使用free与blockerd两个分类，因此需将classifier[6]的输出类别数量调整为2。请在代码下方加入print(model.classifier[6])指令，打印出修改后的内容，会看到out_features的值已经变成2。
在线咨询
