From 85506b439f7d04b6e3aec347e373ea9f1a673282 Mon Sep 17 00:00:00 2001
From: Constantin Wenger <constantin.wenger@googlemail.com>
Date: Tue, 13 Aug 2019 20:38:51 +0000
Subject: [PATCH] make sure duplicate set abbreviations are removed
---
generate_data.py | 24 +++++++++++++-----------
1 files changed, 13 insertions(+), 11 deletions(-)
diff --git a/generate_data.py b/generate_data.py
index c283b97..4131410 100644
--- a/generate_data.py
+++ b/generate_data.py
@@ -1,16 +1,18 @@
-from glob import glob
-import matplotlib.pyplot as plt
-import matplotlib.image as mpimage
-import pickle
-import math
-import random
-import os
import cv2
import fetch_data
+from glob import glob
+import math
+import matplotlib.pyplot as plt
+import matplotlib.image as mpimage
import numpy as np
+import os
import pandas as pd
+import pickle
+import random
import transform_data
+from config import Config
+
class Backgrounds:
"""
@@ -96,8 +98,8 @@
#bg.get_random(display=True)
card_pool = pd.DataFrame()
- for set_name in fetch_data.all_set_list:
- df = fetch_data.load_all_cards_text('data/csv/%s.csv' % set_name)
+ for set_name in Config.all_set_list:
+ df = fetch_data.load_all_cards_text('%s/csv/%s.csv' % (Config.data_dir, set_name))
#for _ in range(3):
# card_info = df.iloc[random.randint(0, df.shape[0] - 1)]
# # Currently ignoring planeswalker cards due to their different card layout
@@ -119,8 +121,8 @@
'''
for _, card_info in card_pool.iterrows():
- img_name = '../usb/data/png/%s/%s_%s.png' % (card_info['set'], card_info['collector_number'],
- fetch_data.get_valid_filename(card_info['name']))
+ img_name = '%s/card_img/png/%s/%s_%s.png' % (Config.data_dir, card_info['set'], card_info['collector_number'],
+ fetch_data.get_valid_filename(card_info['name']))
print(img_name)
card_img = cv2.imread(img_name)
if card_img is None:
--
Gitblit v1.10.0