import csv
import subprocess
import os
runMode = "train"
classes = ["Snowman"]
with open('class-descriptions-boxable.csv', mode='r') as infile:
reader = csv.reader(infile)
dict_list = {rows[1]:rows[0] for rows in reader}
subprocess.run(['rm', '-rf', 'JPEGImages'])
subprocess.run([ 'mkdir', 'JPEGImages'])
subprocess.run(['rm', '-rf', 'labels'])
subprocess.run([ 'mkdir', 'labels'])
for ind in range(0, len(classes)):
className = classes[ind]
print("Class " + str(ind) + " : " + className)
commandStr = "grep " + dict_list[className] + " " + runMode + "-annotations-bbox.csv"
print(commandStr)
class_annotations = subprocess.run(commandStr.split(), stdout=subprocess.PIPE).stdout.decode('utf-8')
class_annotations = class_annotations.splitlines()
totalNumOfAnnotations = len(class_annotations)
print("Total number of annotations : "+str(totalNumOfAnnotations))
cnt = 0
for line in class_annotations[0:totalNumOfAnnotations]:
cnt = cnt + 1
print("annotation count : " + str(cnt))
lineParts = line.split(',')
subprocess.run([ 'aws', 's3', '--no-sign-request', '--only-show-errors', 'cp', 's3://open-images-dataset/'+runMode+'/'+lineParts[0]+".jpg", 'JPEGImages/'+lineParts[0]+".jpg"])
with open('labels/%s.txt'%(lineParts[0]),'a') as f:
f.write(' '.join([str(ind),str((float(lineParts[5]) + float(lineParts[4]))/2), str((float(lineParts[7]) + float(lineParts[6]))/2), str(float(lineParts[5])-float(lineParts[4])),str(float(lineParts[7])-float(lineParts[6]))])+'\n')
python-linux命令處理文件
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.