This action will force synchronization from 头上有煎饺/Defect-Detection-with-tensorflow, which will overwrite any changes that you have made since you forked the repository, and can not be recovered!!!
Synchronous operation will process in the background and will refresh the page when finishing processing. Please be patient.
主要目的是做一个基于图像的胶囊表面缺陷检测,结合opencv和图像处理技术,再通过深度学习训练个模型达到效果
目前问题:对于胶囊值缺了一角或者比较小的裂痕神经网络识别并不好,因为电脑配置不行,没试过VGG这些比较深的模型, 感觉只有用带注意力模块的卷积模型才能分类,因为这个问题其实和区分两个品种不同但是区别很小的动物的问题很像,尴尬的是 我没有能力去实现注意力网络
完成作品的在文件夹FinalVer里面
VideoPart.py:获取摄像头图像,保存训练数据
Data_Trans.py:将图片和标记生成TFRecord格式的数据流
capsule_inspection.py:模型构建,训练和测试,里面虽然写了bn层但是没有使用,使用了bn好像反而会影响结果
我是将每一类的图像都放在了同一个文件夹,区分的方式是文件的名字(文件命名是在VideoPart中完成的)。读取 的时候使用了re模块匹配名称打上对应的标签,后来我觉得分文件夹更好,但是懒得改了。
类别命名方式:完整的胶囊:0_xxxxx.jpg,凹陷的胶囊:1_xxxxx.jpg,缺少身体的:3_xxxxx.jpg,缺少帽子的:4_xxxxx.jpg
每张照片我用的普通1080p的usb摄像头照的,然后截取的227x227大小的地方,实际上就是左上角的那一小块,背景是用临摹板(led 背照的板子)主要是为了消除阴影的影响,我也没测试过阴影影不影响。照相的时候会可以小台灯照一下目标,改变一下光照环境 增强数据。
每一类缺陷我照了3000张图片,最开始我用反转来增强数据,但是效果并不明显,可能是胶囊图像并不复杂,简单的反转增加亮度并 不能消除过拟合和不够泛化的问题,最后我还是手动调整胶囊位置多照几张,因为实在是照得累....于是就只照了3000张,其实每一 类实际照的胶囊数最多500个。
总共迭代了400次,用的自适应梯度下降,虽然不是很光滑的下降曲线,但是很快就是收敛到loss值只有0.2左右,我没有继续训练 不知道过拟合有没有发生,但是测试效果非常好,无论光照变化还是胶囊位置变化检测效果几乎百分之百。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。