added croping and remembering detected cards as well es putting them in a f...

Constantin Wenger

2019-08-10 f9d5508010c4e67e9b1af6bb8347ba2a3023fa78

 setup_train.py

@@ -1,13 +1,14 @@
import os
from glob import glob
import os
import random
import transform_data

from config import Config


def main():
    random.seed()
    data_list = []
    for subdir in glob('%s/train/*' % transform_data.data_dir):
    for subdir in glob('%s/train/*_update' % Config.data_dir):
        for data in glob(subdir + "/*.jpg"):
            data_list.append(os.path.abspath(data))
    random.shuffle(data_list)
@@ -15,10 +16,10 @@
    test_ratio = 0.1
    test_list = data_list[:int(test_ratio * len(data_list))]
    train_list = data_list[int(test_ratio * len(data_list)):]
    with open('%s/train.txt' % transform_data.darknet_dir, 'w') as train_txt:
    with open('%s/train.txt' % Config.darknet_dir, 'w') as train_txt:
        for data in train_list:
            train_txt.write(data + '\n')
    with open('%s/test.txt' % transform_data.darknet_dir, 'w') as test_txt:
    with open('%s/test.txt' % Config.darknet_dir, 'w') as test_txt:
        for data in test_list:
            test_txt.write(data + '\n')
    return