Harshvardhan Chandirasekar
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
import os
|
|
import urllib.request as dowReq
|
|
import zipfile
|
|
|
|
val = input("Enter BDD or COCO :")
|
|
if(val == "COCO"):
|
|
url = "https://cloud.hipert.unimore.it/s/LNxBDk4wzqXPL8c/download"
|
|
lib = "..\demo\COCO_val2017"
|
|
lib_zip = "COCO_val2017.zip"
|
|
elif(val == "BDD"):
|
|
url = "https://cloud.hipert.unimore.it/s/bikqk3FzCq2tg4D/download"
|
|
lib = "..\demo\BDD100k_val"
|
|
lib_zip = "BDD100k_val.zip"
|
|
|
|
dowReq.urlretrieve(url,lib_zip)
|
|
|
|
with zipfile.ZipFile(lib_zip,'r') as zip_ref:
|
|
zip_ref.extractall(lib)
|
|
|
|
labelFolder = lib + "\labels"
|
|
imageFolder = lib + "\images"
|
|
|
|
file1 = open(".\\..\\demo\\all_labels.txt","a")
|
|
path1 = os.path.realpath(labelFolder)
|
|
for file in os.listdir(labelFolder):
|
|
valTemp = path1 + "\\" + file
|
|
valTemp = valTemp + " \n"
|
|
file1.write(valTemp)
|
|
file1.close()
|
|
|
|
file2 = open(".\\..\\demo\\all_images.txt","a")
|
|
path2 = os.path.realpath(imageFolder)
|
|
for file in os.listdir(imageFolder):
|
|
pathtemp = path2 + "\\" + file
|
|
pathtemp = pathtemp + " \n"
|
|
file2.write(pathtemp)
|
|
file2.close()
|
|
|
|
print("Completed")
|