前言
今天放松一下,随便看看这个YOLOV5 的识别部分的代码是怎么做的,先前的话我们自己手动实现了一个非常简易的分类框架,HuClassfiy(已经上传Gitee,方便各位访问),那么这里的话想要使用YOLOV5做点好玩的,也必须要对整个的代码流程进行梳理。原理就不用说了,老复杂了,所以先从简单的来探索。
我们原来的实现这个detect的代码非常简单,后面会贴出,我注释后的detect代码
import argparse
from PIL import Image
from utils.DataSet.MyDataSet import MyDataSet