dataset.py 965 B

12345678910111213141516171819202122232425
  1. # Copyright (c) 2021 NVIDIA Corporation. All rights reserved.
  2. import gdown
  3. import os
  4. ## VOC dataset
  5. url0 = 'http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar'
  6. # if the url0 above fails, uncomment the url0 below and rebiuld your container
  7. #url0 = 'https://drive.google.com/file/d/1YkakD_qJuKmwAZF_HwLGLNb8fcqZbtx7/view?usp=sharing'
  8. output_python0 = '/workspace/tlt-experiments/data/VOCtrainval_11-May-2012.tar'
  9. ## Kitti detection dataset for yolov4
  10. url1 = 'http://www.cvlibs.net/download.php?file=data_object_label_2.zip'
  11. url2 = 'http://www.cvlibs.net/download.php?file=data_object_image_2.zip'
  12. output_python1 = '/workspace/tlt-experiments/data/data_object_label_2.zip'
  13. output_python2 = '/workspace/tlt-experiments/data/data_object_image_2.zip'
  14. gdown.download(url0, output_python0, quiet=False, proxy=None)
  15. gdown.download(url1, output_python1, quiet=False, proxy=None)
  16. gdown.download(url2, output_python2, quiet=False, proxy=None)