dmlc--dgl
Tong He
cbee427839
* add working scripts * add frcnn training script * remove redundent files * refactor validation computation, will optimize sgdet and training * validation finally finished * f-rcnn training * test reldn * rm file * update reldn training * data preprocess to h5 * temp * use coco json * fix conflict * new obj dataset for detection * update training * before cleanup * remove abundant files * add arg parse to train * cleanup code file * update * fix * add readme * add ipynb as demo * add demo pic * update readme * add demo script * improve paths * improve readme * add docstrings * fix args description * update readme * add models from s3 * update README Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>
47 行
2.2 KiB
Python
47 行
2.2 KiB
Python
"""Pascal VOC object detection dataset."""
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
import os
|
|
import logging
|
|
import warnings
|
|
import json
|
|
import pickle
|
|
import numpy as np
|
|
import mxnet as mx
|
|
from gluoncv.data import COCODetection
|
|
from collections import Counter
|
|
|
|
class VGObject(COCODetection):
|
|
CLASSES = ["airplane", "animal", "arm", "bag", "banana", "basket", "beach",
|
|
"bear", "bed", "bench", "bike", "bird", "board", "boat", "book",
|
|
"boot", "bottle", "bowl", "box", "boy", "branch", "building", "bus",
|
|
"cabinet", "cap", "car", "cat", "chair", "child", "clock", "coat",
|
|
"counter", "cow", "cup", "curtain", "desk", "dog", "door", "drawer",
|
|
"ear", "elephant", "engine", "eye", "face", "fence", "finger", "flag",
|
|
"flower", "food", "fork", "fruit", "giraffe", "girl", "glass", "glove",
|
|
"guy", "hair", "hand", "handle", "hat", "head", "helmet", "hill",
|
|
"horse", "house", "jacket", "jean", "kid", "kite", "lady", "lamp",
|
|
"laptop", "leaf", "leg", "letter", "light", "logo", "man", "men",
|
|
"motorcycle", "mountain", "mouth", "neck", "nose", "number", "orange",
|
|
"pant", "paper", "paw", "people", "person", "phone", "pillow", "pizza",
|
|
"plane", "plant", "plate", "player", "pole", "post", "pot", "racket",
|
|
"railing", "rock", "roof", "room", "screen", "seat", "sheep", "shelf",
|
|
"shirt", "shoe", "short", "sidewalk", "sign", "sink", "skateboard",
|
|
"ski", "skier", "sneaker", "snow", "sock", "stand", "street",
|
|
"surfboard", "table", "tail", "tie", "tile", "tire", "toilet",
|
|
"towel", "tower", "track", "train", "tree", "truck", "trunk",
|
|
"umbrella", "vase", "vegetable", "vehicle", "wave", "wheel",
|
|
"window", "windshield", "wing", "wire", "woman", "zebra"]
|
|
def __init__(self, **kwargs):
|
|
super(VGObject, self).__init__(**kwargs)
|
|
|
|
@property
|
|
def annotation_dir(self):
|
|
return ''
|
|
|
|
def _parse_image_path(self, entry):
|
|
dirname = 'VG_100K'
|
|
filename = entry['file_name']
|
|
abs_path = os.path.join(self._root, dirname, filename)
|
|
return abs_path
|