From 7018d163ed2741bc1f095f9f129368beecdee567 Mon Sep 17 00:00:00 2001 From: hchandirasekar Date: Fri, 26 Mar 2021 19:37:44 +0530 Subject: [PATCH] python download file --- scripts/download_validation.py | 40 ++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 scripts/download_validation.py diff --git a/scripts/download_validation.py b/scripts/download_validation.py new file mode 100644 index 0000000..e49f2ab --- /dev/null +++ b/scripts/download_validation.py @@ -0,0 +1,40 @@ +import os +from pathlib import Path +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") \ No newline at end of file