Bob's Blog

Web开发、测试框架、自动化平台、APP开发、机器学习等


google colab的一些使用限制

目前有需求是训练自定义的数据集,用了yolov4 darknet,但是公司发的老版mac训练起来实在是慢且发烫,于是打算试一试google的colab,用一下免费的GPU。

先新建了一个notebook,选择菜单


YOLOV5根据屏幕图像实时给出预测结果

目前在用yolov5,在调试时需要检验当前模型是否匹配出了期望的实体类型。因为目前需要匹配的就是当前计算机显示出的各类图像,于是在想做一个实时的检测屏幕图像并给出预测结果,就像之前文章中的根据摄像头实时给出预测一样。(不过有个问题,记


YOLO v5 训练自定义数据集

Darknet有两个版本,分别对应着v3和v4;在v4后不久,就有了pytorch版本的v5,据称是性能和效果更好。这里主要是翻译如何用yolo v5来训练自定义的数据集,加了些自己尝试的过程,参考了三篇文章,链接见文章末尾。