Skip to content

Commit

Permalink
Update __init__.py
Browse files Browse the repository at this point in the history
  • Loading branch information
thucbx99 committed Aug 1, 2023
1 parent c0bb105 commit 8060093
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions tllib/vision/datasets/object_detection/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,32 +43,32 @@ def __init__(self, root, split="trainval", year=2007, ext='.jpg', download=True)


class VOC2007(VOCBase):
archive_name = 'VOC2007.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/07c03846908e432e87e6/?dl=1'
archive_name = 'VOC2007.tgz'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/800a9495d3b74612be3f/?dl=1'

def __init__(self, root):
super(VOC2007, self).__init__(root)


class VOC2012(VOCBase):
archive_name = 'VOC2012.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/d9c348407e4946068e89/?dl=1'
archive_name = 'VOC2012.tgz'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/a7e7ab88f727408eaf32/?dl=1'

def __init__(self, root):
super(VOC2012, self).__init__(root, year=2012)


class VOC2007Test(VOCBase):
archive_name = 'VOC2007.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/07c03846908e432e87e6/?dl=1'
archive_name = 'VOC2007.tgz'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/800a9495d3b74612be3f/?dl=1'

def __init__(self, root):
super(VOC2007Test, self).__init__(root, year=2007, split='test')


class Clipart(VOCBase):
archive_name = 'clipart.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/6b8c45c5b7f34272baee/?dl=1'
archive_name = 'clipart.zip'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/c853a66786e2416a8f18/?dl=1'


class VOCPartialBase:
Expand All @@ -88,40 +88,40 @@ def __init__(self, root, split="trainval", year=2007, ext='.jpg', download=True)


class VOC2007Partial(VOCPartialBase):
archive_name = 'VOC2007.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/07c03846908e432e87e6/?dl=1'
archive_name = 'VOC2007.tgz'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/800a9495d3b74612be3f/?dl=1'

def __init__(self, root):
super(VOC2007Partial, self).__init__(root)


class VOC2012Partial(VOCPartialBase):
archive_name = 'VOC2012.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/d9c348407e4946068e89/?dl=1'
archive_name = 'VOC2012.tgz'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/a7e7ab88f727408eaf32/?dl=1'

def __init__(self, root):
super(VOC2012Partial, self).__init__(root, year=2012)


class VOC2007PartialTest(VOCPartialBase):
archive_name = 'VOC2007.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/07c03846908e432e87e6/?dl=1'
archive_name = 'VOC2007.tgz'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/800a9495d3b74612be3f/?dl=1'

def __init__(self, root):
super(VOC2007PartialTest, self).__init__(root, year=2007, split='test')


class WaterColor(VOCPartialBase):
archive_name = 'watercolor.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/d2d321affb5f466aae5b/?dl=1'
archive_name = 'watercolor.zip'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/9f322fd8496f4766ad93/?dl=1'

def __init__(self, root):
super(WaterColor, self).__init__(root, split='train')


class WaterColorTest(VOCPartialBase):
archive_name = 'watercolor.tar'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/d2d321affb5f466aae5b/?dl=1'
archive_name = 'watercolor.zip'
dataset_url = 'https://cloud.tsinghua.edu.cn/f/9f322fd8496f4766ad93/?dl=1'

def __init__(self, root):
super(WaterColorTest, self).__init__(root, split='test')
Expand Down

0 comments on commit 8060093

Please sign in to comment.