yolo和yolo-tiny实际使用对比

最近在学习使用yolo进行目标识别,然而由于显存问题,yolo官方的train.py跑不了,所以尝试了两种方法,一种是减小yolo训练时的batch_size,另一种是使用yolo-tiny,现记录一下两者的效果对比。 先交代一下前提,我的显卡是GTX1050Ti的,显存4G,训练只有一种类别,训练集有140张图,平均每张图上有3个左右目标。使用的是yolov3。 首先使用yolo-tiny,ba
相关文章
相关标签/搜索