本文对TensorRT对深度学习模型的优化进行介绍,主要是翻译英文文档。
本文给出了一个分析图像的主题颜色,或者说是主要颜色的方法和实现。它可以提取一幅图像中,比如服装,的主要颜色。
YOLOv5的推理与输出层,涵盖模型信息、grid、loss、设备端移植等,尤其是build_targets函数进行了超详细的注释。
从代码的角度熟悉nn.Module,跳过苦涩难懂的文档。Module的代码非常简单,比网上各种文档简单多了。
YOLOv5的网络结构展示,结合分析代码实现,使用Netron进行可视化。
第一次认识知识蒸馏是从模型压缩的角度,使用泛化能力更强的大模型(如Resnet-50),蒸馏出一个小模型(Resnet-18),提升小模型的泛化能力。它的思辨和原理是什么,论文《Distilling the Knowledge in a Neural Network》,给出了答案。为了表达出该技术的思辨过程,尽量保留原文意思。