fetch_girl_images.py 文件源码

def download_proprocess_dataset():
    csvfile = open(GIRL_MARK_FILE, "r")
    reader = csv.reader(csvfile)
    train_data_arr = []
    test_data_arr = []
    for index, item in enumerate(reader):
        if index % 5 == 0:
            image_filename = download_girl_image(item[0], IMG_TEST_DIR)
            if imghdr.what(image_filename) is not None:
                img = process_image(image_filename)
                test_data_arr.append(mg.mark_girl(img, int(item[1])))
        else:
            image_filename = download_girl_image(item[0], IMG_TRAIN_DIR)
            if imghdr.what(image_filename) is not None:
                img = process_image(image_filename)
                train_data_arr.append(mg.mark_girl(img, int(item[1])))
    writer = tf.python_io.TFRecordWriter(mg.FILE_NAME_TEST)
    for data in test_data_arr:
        writer.write(data)
    writer.close()
    writer = tf.python_io.TFRecordWriter(mg.FILE_NAME_TRAIN)
    for data in train_data_arr:
        writer.write(data)
    writer.close()