HelloSunny123321的博客

这是我训练一个垃圾分类的yolo-tiny网络的输出可视化结果，怀疑可能有如下几条原因导致了如此波动： 1、数据集问题。大部分图片都是从网络上爬取的，少部分是自己拍摄的，质量层次不齐且无法充分表现出此类垃圾特征。因为已经检查过直接用来训练的txt文件，所以可以基本排除是格式转化出现的问题。 2、learning rate选择不合适。 batch=64 subdivisions=8 width=...

def imShow(path): import cv2 import matplotlib.pyplot as plt %matplotlib inline image = cv2.imread(path) height, width = image.shape[:2] resized_image = cv2.resize(image,(3*width, 3*heigh...