From 62084d6733a78822f39dd8a1fe01c77eea281a65 Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Sat, 08 Sep 2018 04:11:49 +0000
Subject: [PATCH] More testing stuffs
---
transform_data.py | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/transform_data.py b/transform_data.py
index 867abaf..a05bc89 100644
--- a/transform_data.py
+++ b/transform_data.py
@@ -14,7 +14,7 @@
from imgaug import parameters as iap
card_mask = cv2.imread('data/mask.png')
-data_dir = os.path.abspath('/media/edmond/My Passport/data')
+data_dir = os.path.abspath('/media/win10/data')
darknet_dir = os.path.abspath('darknet')
@@ -529,15 +529,15 @@
])
if i % 3 == 0:
generator.generate_non_obstructive()
- generator.export_training_data(visibility=0.0, out_name='%s/train/non_obstructive/%s_%d'
+ generator.export_training_data(visibility=0.0, out_name='%s/train/non_obstructive_skew/%s_%d'
% (data_dir, out_name, j), aug=seq)
elif i % 3 == 1:
generator.generate_horizontal_span(theta=random.uniform(-math.pi, math.pi))
- generator.export_training_data(visibility=0.0, out_name='%s/train/horizontal_span/%s_%d'
+ generator.export_training_data(visibility=0.0, out_name='%s/train/horizontal_span_skew/%s_%d'
% (data_dir, out_name, j), aug=seq)
else:
generator.generate_vertical_span(theta=random.uniform(-math.pi, math.pi))
- generator.export_training_data(visibility=0.0, out_name='%s/train/vertical_span/%s_%d'
+ generator.export_training_data(visibility=0.0, out_name='%s/train/vertical_span_skew/%s_%d'
% (data_dir, out_name, j), aug=seq)
#generator.generate_horizontal_span(theta=random.uniform(-math.pi, math.pi))
#generator.render(display=True, aug=seq, debug=True)
--
Gitblit v1.10.0