Skip to content

Commit de38521

Browse files
committed
Update COCODetection
1 parent 0628cb0 commit de38521

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

data/coco.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,13 @@
1+
import os
2+
import os.path
3+
import sys
4+
import torch
5+
import torch.utils.data as data
6+
import torchvision.transforms as transforms
7+
import cv2
8+
import numpy as np
9+
10+
111
class COCOAnnotationTransform(object):
212
"""Transforms a VOC annotation into a Tensor of bbox coords and label index
313
Initilized with a dictionary lookup of classnames to indexes

0 commit comments

Comments
 (0)