search

wusaifei / garbage_classify

本文新增添分类，检测，换脸技术等学习教程，各种调参技巧和tricks，卷积结构详细解析可视化，注意力机制代码等详解！本次垃圾分类挑战杯，目的在于构建基于深度学习技术的图像分类模型，实现垃圾图片类别的精准识别，大赛参考深圳垃圾分类标准，按可回收物、厨余垃圾、有害垃圾和其他垃圾四项分类。本项目包含完整的分类网络、数据增强、SVM等各种分类增强策略，后续还会继续更新新的分类技巧。
676 stars 176 forks source link

请教run.py出现valueError， #27

Open mars918 opened 4 years ago

mars918 commented 4 years ago

ValueError: Error when checking input: expected conv2d_1_input to have 4 dimensions, but got array with shape (16, 1)

wusaifei commented 4 years ago

你好，是不是你的图像本身的问题呢。这个意思是输入到网络的图像和目标的shape不对应导致的。