本文对TensorRT sampleONNXMNIST进行了介绍和实践,与官方无异,但其中有一些实践经验值得参考。
本文对TensorRT对深度学习模型的优化进行介绍,主要是翻译英文文档。
本文给出了一个分析图像的主题颜色,或者说是主要颜色的方法和实现。它可以提取一幅图像中,比如服装,的主要颜色。
YOLOv5的推理与输出层,涵盖模型信息、grid、loss、设备端移植等,尤其是build_targets函数进行了超详细的注释。
从代码的角度熟悉nn.Module,跳过苦涩难懂的文档。Module的代码非常简单,比网上各种文档简单多了。
YOLOv5的网络结构展示,结合分析代码实现,使用Netron进行可视化。