[PyTorch] torchvision.dataset.CoCoDetection

dataLoader를 만들어서 train함수에 넣어주어야 하는데 이 때 dataLoader를 만들려면 class로 데이터셋에 대한 정의가 있어야 한다.

 

 

!git clone https://github.com/pdollar/coco/
%cd coco/PythonAPI/
!make
!python setup.py install
%cd ../..
!rm -r coco
!sudo apt-get update
!sudo apt-get install aria2

!aria2c -x 10 -j 10 http://images.cocodataset.org/zips/train2017.zip
!aria2c -x 10 -j 10 http://images.cocodataset.org/zips/val2017.zip
!aria2c -x 10 -j 10 http://images.cocodataset.org/annotations/annotations_trainval2017.zip

!unzip annotations_trainval2017.zip
!unzip train2017.zip
!unzip val2017.zip

path2data="/train2017"
path2json="/annotations/instances_train2017.json"
coco_train = torchvision.datasets.CocoDetection(root = path2data,
                                annFile = path2json)

 

 

 

PyTorch torchvision COCO Dataset