| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- import os
- import cv2
- import numpy as np
- import utils
- def argsProcessor():
- import argparse
- parser = argparse.ArgumentParser()
- parser.add_argument("-i", "--dataPath", help="DataPath")
- parser.add_argument("-o", "--outputFiles", help="outputFiles", default="bar")
- return parser.parse_args()
- args = argsProcessor()
- output_dir = args.outputFiles
- if (not os.path.isdir(output_dir)):
- os.mkdir(output_dir)
- dir = args.dataPath
- import csv
- with open(output_dir+"/gt.csv", 'a') as csvfile:
- spamwriter_1 = csv.writer(csvfile, delimiter=',',
- quotechar='|', quoting=csv.QUOTE_MINIMAL)
- for image in os.listdir(dir):
- if image.endswith("jpg"):
- if os.path.isfile(dir+"/"+image+".csv"):
- with open(dir+"/"+image+ ".csv", 'r') as csvfile:
- spamwriter = csv.reader(csvfile, delimiter=' ',
- quotechar='|', quoting=csv.QUOTE_MINIMAL)
- img = cv2.imread(dir +"/"+ image)
- print (image)
- gt= []
- for row in spamwriter:
- gt.append(row)
- # img = cv2.circle(img, (int(float(row[0])), int(float(row[1]))), 2,(255,0,0),90)
- gt =np.array(gt).astype(np.float32)
- # print gt
- gt = gt / (img.shape[1], img.shape[0])
- gt = gt * (1080, 1080)
- img = cv2.resize(img, ( 1080,1080))
- # for a in range(0,4):
- # img = cv2.circle(img, tuple((gt[a].astype(int))), 2, (255, 0, 0), 9)
- # cv2.imwrite("asda.jpg", img)
- # 0/0
- for angle in range(0,271,90):
- img_rotate, gt_rotate = utils.rotate(img, gt, angle)
- for random_crop in range(0,16):
- img_list, gt_list = utils.getCorners(img_rotate, gt_rotate)
- for a in range(0,4):
- print (gt_list[a])
- gt_store = list(np.array(gt_list[a])/(300,300))
- img_store = cv2.resize(img_list[a], (64,64))
- print (tuple(list(np.array(gt_store)*64)))
- # cv2.circle(img_store, tuple(list((np.array(gt_store)*64).astype(int))), 2, (255, 0, 0), 2)
- cv2.imwrite( output_dir+"/"+image + str(angle) +str(random_crop) + str(a) +".jpg", img_store)
- spamwriter_1.writerow(( image + str(angle) +str(random_crop) + str(a) +".jpg", tuple(gt_store)))
|
