OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Create get_openimages_dataset.py 

Script that downloads images with selected classes from OpenImages dataset and create labels in "yolo format"
This commit is contained in:
salbatron 2018-05-17 15:22:49 +02:00 committed by GitHub
parent fb9fcfb3ba
commit 0ecbeeff42
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
scripts/get_openimages_dataset.py Normal file
View File

@ -0,0 +1,19 @@
import csv
import os
#select classes you want to download at https://github.com/openimages/dataset/blob/master/dict.csv
CLASS_LIST = ('/m/01g317','/m/04yx4')
img_name = "111111111111"
#download csv from https://storage.googleapis.com/openimages/web/download.html
with open('path\\train-annotations-bbox.csv', newline='') as csvfile:
bboxs = csv.reader(csvfile, delimiter=',', quotechar='|')
for bbox in bboxs:
if bbox[2] in CLASS_LIST:
if img_name != bbox[0]:
if not os.path.isfile("destination_path\\%s.jpg"%bbox[0]):
os.system("gsutil cp gs://open-images-dataset/train/%s.jpg destination_path"%bbox[0])
out_file = open("destination_path\\%s.txt"%bbox[0], 'w')
img_name = bbox[0]
if img_name == bbox[0]:
out_file.write(str(CLASS_LIST.index(bbox[2])) + " " + str(float(bbox[4])+(float(bbox[5])-float(bbox[4]))/2) + " " + str(float(bbox[6])+(float(bbox[7])-float(bbox[6]))/2)+ " " + str(float(bbox[5])-float(bbox[4])) + " " + str(float(bbox[7])-float(bbox[6])) + '\n')