怎么下载VGG代码、创建模型主要参考了这篇博客https://blog.csdn.net/weixin_42795611/article/details/91126616
数据集是我自己用labelimg创建的，一共9类
但是在运行VGG16.py出现如下问题：
在网上搜了一圈也没找到解决办法，后来把代码重头看一遍，原来是下面最后这行代码出了问题，因为自己的训练种类9类，要把softmax前面对应的输出也写成9，否则就会出现像上面这样维度不匹配的问题
相关文章：
相似解决方案
热门标签