预训练模型预测

本次我主要是利用pytorch自带的一些网络来进行分类任务,也就是化身为调包侠。

  1. 预测单张图像

  • 载入模型

  • 图像预处理

  • 图像载入

  • 执行分类

  • 预测结果分析

  • 将分类结果标注在原图上

  • 结果可视化与输出

  1. 预测视频文件(与前面大同小异)

  1. 预测摄像头实时画面(主要是桢的得到)

参考

github.com/TommyZihao/Train_Custom_Dataset

同济子豪兄

你可能感兴趣的:(深度学习,人工智能,计算机视觉)