Oct, 2023

QWID:量化杂草识别深度神经网络

TL;DR本文提出了一种有效的农业杂草分类解决方案,该方案在推断时优化了模型性能,同时尊重农业领域的约束。我们通过使用 8 位整数(int8)量化的量化深度神经网络模型对包含 9 个杂草类别的数据集进行分类,这与标准的 32 位浮点(fp32)模型有所不同。通过在深度杂草数据集上应用迁移学习和微调,我们将此方法在 ResNet-50 和 InceptionV3 架构上进行了评估,并将其性能与其 int8 量化版本进行了比较。结果显示,在保持准确性的同时,模型大小和推断时间大幅减少,适用于台式机、移动设备和树莓派等实际生产场景。我们的工作为农业中的高效人工智能指明了一个有前景的方向,具有更广泛的应用潜力。