Skip to content

Instantly share code, notes, and snippets.

@aakarsh7599

aakarsh7599/visualize.py

Last active August 2, 2020 15:52
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save aakarsh7599/eb02687c3d9d515ad4219a0a5a82ce4d to your computer and use it in GitHub Desktop.
Save aakarsh7599/eb02687c3d9d515ad4219a0a5a82ce4d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
visualize.py
#Visualizing the Train Dataset
dataset_dicts = get_board_dicts("Text_Detection_Dataset_COCO_Format/train")
#Randomly choosing 3 images from the Set
for d in random.sample(dataset_dicts, 3):
img = cv2.imread(d["file_name"])
visualizer = Visualizer(img[:, :, ::-1], metadata=board_metadata)
vis = visualizer.draw_dataset_dict(d)
cv2_imshow(vis.get_image()[:, :, ::-1])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment