tvm_book.data.imagenet 源代码


[文档]
class ImageNet1kAttr(object):
    def __init__(self):


[文档]
        self.num_class = 1000




[文档]
        self.synset = []




[文档]
        self.classes = []




[文档]
        self.classes_long = []



        for syn, cls, cls_l in _ILSVRC2012_Attr:
            self.synset.append(syn)
            self.classes.append(cls)
            self.classes_long.append(cls_l)