2501_94828736的博客

原创 基于MobileNetV2的猫狗分类实战

本次实战使用的是Kaggle精简版猫狗图像分类数据集，与经典的25000张大型数据集不同，该数据集经过筛选后包含697张标注图片（猫349张，狗348张），分为训练集和测试集两个子目录，标签通过文件夹名隐式标注（ train 目录下为猫、 test 目录下为狗，实际实战中也可按文件名标注，如 cat_001.jpg / dog_001.jpg ）置信度优化：将猫的置信度改为 1-pred ，狗的置信度为 pred ，避免猫的置信度显示为0.1（实际模型认为是猫的概率为0.9）的歧义。