From 46f2b05df318f0d699c6e22df156d5f96b07c65c Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Wed, 05 Sep 2018 21:37:38 +0000
Subject: [PATCH] Training data generation, setting up YOLOv3 using darknet
---
fetch_data.py | 10 +++++++---
1 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/fetch_data.py b/fetch_data.py
index 2d66e02..adad979 100644
--- a/fetch_data.py
+++ b/fetch_data.py
@@ -5,7 +5,7 @@
import re
import os
-all_set_list = ['8ed', 'mrd', 'dst', '5dn', 'chk', 'bok', 'sok', '9ed', 'rav', 'gpt', 'dis', 'csp', 'tsp', 'plc', 'fut',
+all_set_list = ['mrd', 'dst', '5dn', 'chk', 'bok', 'sok', 'rav', 'gpt', 'dis', 'csp', 'tsp', 'plc', 'fut',
'10e', 'lrw', 'mor', 'shm', 'eve', 'ala', 'con', 'arb', 'm10', 'zen', 'wwk', 'roe', 'm11', 'som', 'mbs',
'nph', 'm12', 'isd', 'dka', 'avr', 'm13', 'rtr', 'gtc', 'dgm', 'm14', 'ths', 'bng', 'jou']
@@ -84,15 +84,19 @@
def main():
+ '''
for set_name in all_set_list:
csv_name = 'data/csv/%s.csv' % set_name
if not os.path.isfile(csv_name):
- df = fetch_all_cards_text(url='https://api.scryfall.com/cards/search?q=layout:normal+set:%s+lang:en' % set_name,
- csv_name=csv_name)
+ df = fetch_all_cards_text(url='https://api.scryfall.com/cards/search?q=layout:normal+set:%s+lang:en+frame:2003'
+ % set_name, csv_name=csv_name)
else:
df = load_all_cards_text(csv_name)
print(csv_name)
fetch_all_cards_image(df, out_dir='../usb/data/png/%s' % set_name)
+ '''
+ df = fetch_all_cards_text(url='https://api.scryfall.com/cards/search?q=layout:normal+lang:en+frame:2003',
+ csv_name='data/csv/all.csv')
pass
--
Gitblit v1.10.0