深度学习
文章平均质量分 95
小虚无
心静自然凉
展开
-
AMFF-YOLOX:基于YOLOX的工业缺陷检测注意机制与多特征融合研究(包含论文讲解、代码实现和可运行的源代码)
工业缺陷检测在产品质量改进中具有重要意义,深度学习方法是目前的主流方法。然而工业产品的体积巨大,主流探测器在快速检测过程中无法保持较高的准确率。为了解决上述问题,本文提出了一种基于YOLOX的改进的工业缺陷检测器AMFF-YOLOX。该方法可以减少主干网络中瓶颈的激活函数和归一化操作,并在特征提取网络中添加注意机制和自适应空间特征融合,使网络能够更好地关注对象。最终,在网络预测速度的前提下提高了预测的准确性,与主流检测器相比具有良好的性能。原创 2023-10-19 15:45:33 · 303 阅读 · 3 评论 -
Pytorch复现经典扩散模型DDPM&DDIM&PLMS及分布式训练应用
当前,生成式人工智能(AIGC)大模型已被越来越广泛应用在工业、动漫业、设计业等诸多场景。本文根据经典扩散模型DDPM、DDIM和PLMS进行了代码层面的详细解读和复现,并实现了多机多卡分布式训练。原创 2023-08-15 03:19:13 · 4823 阅读 · 42 评论