-
Notifications
You must be signed in to change notification settings - Fork 74
/
make_voc_list.py
executable file
·38 lines (28 loc) · 1.04 KB
/
make_voc_list.py
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
27
28
29
30
31
32
33
34
35
36
37
38
import os
import re
import numpy as np
import sys
import argparse
import skimage
def main(train_file: str, output_file: str):
image_path_list = np.loadtxt(train_file, dtype=str)
if not os.path.exists('data'):
os.makedirs('data')
ann_list = list(image_path_list)
ann_list = [re.sub(r'JPEGImages', 'labels', s) for s in ann_list]
ann_list = [re.sub(r'.jpg', '.txt', s) for s in ann_list]
lines = np.array([
np.array([
image_path_list[i],
np.loadtxt(ann_list[i], dtype=float, ndmin=2),
np.array(skimage.io.imread(image_path_list[i]).shape[0:2])]
) for i in range(len(ann_list))])
np.save(output_file, lines)
def parse_arguments(argv):
parser = argparse.ArgumentParser()
parser.add_argument('train_file', type=str, help='trian.txt file path')
parser.add_argument('output_file', type=str, help='output file path')
return parser.parse_args(argv)
if __name__ == "__main__":
args = parse_arguments(sys.argv[1:])
main(args.train_file, args.output_file)