pycocotools coco格式数据集可视化
2021/6/10 18:32:25
本文主要是介绍pycocotools coco格式数据集可视化,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
pycocotools coco格式数据集可视化
import cv2 import os import numpy as np from pycocotools.coco import COCO colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0), (255, 0, 255), (0, 255, 255)] img_path = 'train/images' annFile = 'train/annotations/train.json' save_path = 'train/images_vis' if not os.path.exists(save_path): os.makedirs(save_path) def draw_rectangle(coordinates, image, image_name): for coordinate in coordinates: left, top, right, bottom, label = map(int, coordinate) color = colors[label % len(colors)] cv2.rectangle(image, (left, top), (right, bottom), color, 2) cv2.putText(image, str(label), (left, top), cv2.FONT_HERSHEY_SIMPLEX, 1.2, color, 2) cv2.imwrite(save_path + '/' + image_name, image) coco = COCO(annFile) # catIds = coco.getCatIds(catNms=['Crack','Manhole', 'Net', 'Pothole','Patch-Crack', "Patch-Net", "Patch-Pothole", "other"]) # catIds = coco.getCatIds() # imgIds = coco.getImgIds(catIds=catIds) imgIds = coco.getImgIds() for imgId in imgIds: img = coco.loadImgs(imgId)[0] image_name = img['file_name'] annIds = coco.getAnnIds(imgIds=img['id'], catIds=[], iscrowd=None) anns = coco.loadAnns(annIds) # coco.showAnns(anns) coordinates = [] img_raw = cv2.imread(os.path.join(img_path, image_name)) for j in range(len(anns)): coordinate = anns[j]['bbox'] coordinate[2] += coordinate[0] coordinate[3] += coordinate[1] coordinate.append(anns[j]['category_id']) coordinates.append(coordinate) draw_rectangle(coordinates, img_raw, image_name)
这篇关于pycocotools coco格式数据集可视化的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-07-05feign默认connecttimeout和readtimeout是多少-icode9专业技术文章分享
- 2024-07-05idea控制台,日志太多,导致部分想看得日志被刷走 搜不到-icode9专业技术文章分享
- 2024-07-05The server selected protocol version Tls10 is not accepted by client preferences [TLs12]-icode9专业技术文章分享
- 2024-07-05怎么清理项目缓存-icode9专业技术文章分享
- 2024-07-04安装 Eyoucms详细图文教程-icode9专业技术文章分享
- 2024-07-04ueditor 复制文章时,图片的链接是一个下载图片地址,该如何处理?-icode9专业技术文章分享
- 2024-07-04怎样判断host有没有对wordpress有缓存呢-icode9专业技术文章分享
- 2024-07-04具有编译功能的系统make后,无法ssh连接-icode9专业技术文章分享
- 2024-07-04make后如何升级ssh-icode9专业技术文章分享
- 2024-07-03微信支付提示下单账户与支付账户不一致-icode9专业技术文章分享