# 创作灵感
最近也是自己在弄目标检测方面的东西,然后这也是自己碰到的问题,想着能分享一下,希望对有需要的人有所帮助。也欢迎大家来讨论问题、交流心得!!!
查了许多基本上(应该就是没有啦)找不到一个画YIOLOv8原始框的代码呀,所以就尝试查看YOLOv8的原始代码来绘制原始框。
满足需求请给我点个赞呀,引用望标注转载!!!
第一章、源码
不废话,先把源码给你们,不会改代码的友友可以看我第三章的讲解呀!!!
我代码习惯挺好的,嘿嘿,应该都可以看懂!!!(好像废话有点多呀~~~)
import os
import cv2
# 不要动这个颜色!!!!!!!!!!!!!! 直接划到下面的 if __name__ == "__main__": 里进行修改 其他部分不要动!!!!
colors = [(56, 56, 255), (151, 157, 255),(31, 112, 255),(29, 178, 255),(49, 210, 207), (10, 249, 72), (23, 204, 146),
(134, 219, 61),(52, 147, 26), (187, 212, 0), (168, 153, 44),(255, 194, 0), (147, 69, 52), (255, 115, 100),
(236, 24, 0), (255, 56, 132), (133, 0, 82), (255, 56, 203),(200, 149, 255),(199, 55, 255)]
class Annotator:
def __init__(self, im, line_width=None):
"""Initialize the Annotator class with image and line width."""
assert im.data.contiguous, 'Image not contiguous.