Hugging Face's logo

Spaces:
Realcat
/
image-matching-webui
Running

App Files Community
3
image-matching-webui
File size: 882 Bytes 
a80d6bb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
import os
import argparse

def prepare_coco(args):
    train_file = open(os.path.join(args.saved_dir, args.saved_txt), 'w')
    dirs = os.listdir(args.raw_dir)

    for file in dirs:
        # Write training files
        train_file.write('%s\n' % (file))

    print('Data Preparation Finished.')

if __name__ == '__main__':
    arg_parser = argparse.ArgumentParser(description="coco prepareing.")
    arg_parser.add_argument('--dataset', type=str, default='coco',
                             help='')
    arg_parser.add_argument('--raw_dir', type=str, default='',
                             help='')
    arg_parser.add_argument('--saved_dir', type=str, default='',
                             help='')
    arg_parser.add_argument('--saved_txt', type=str, default='train2017.txt',
                             help='')
    args = arg_parser.parse_args() 

    prepare_coco(args)