|
|
|
|
|
import argparse
|
|
|
|
|
|
import mmengine
|
|
|
|
|
|
|
|
|
def parse_args():
|
|
|
parser = argparse.ArgumentParser(
|
|
|
description='Convert txt annotation list to json')
|
|
|
parser.add_argument(
|
|
|
'annofile', type=str, help='the txt annotation file to convert')
|
|
|
parser.add_argument(
|
|
|
'--format',
|
|
|
type=str,
|
|
|
default='rawframes',
|
|
|
choices=['rawframes', 'videos'],
|
|
|
help='the format of the txt annotation file')
|
|
|
parser.add_argument(
|
|
|
'--output',
|
|
|
type=str,
|
|
|
default=None,
|
|
|
help=(
|
|
|
'the output file name, use annofile.replace(\'.txt\', \'.json\') '
|
|
|
'if the arg value is None'))
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
return args
|
|
|
|
|
|
|
|
|
def lines2dictlist(lines, format):
|
|
|
"""Convert lines in 'txt' format to dictionaries in 'json' format.
|
|
|
Currently support single-label and multi-label.
|
|
|
|
|
|
Example of a single-label rawframes annotation txt file:
|
|
|
|
|
|
.. code-block:: txt
|
|
|
|
|
|
(frame_dir num_frames label)
|
|
|
some/directory-1 163 1
|
|
|
some/directory-2 122 1
|
|
|
some/directory-3 258 2
|
|
|
|
|
|
Example of a multi-label rawframes annotation txt file:
|
|
|
|
|
|
.. code-block:: txt
|
|
|
|
|
|
(frame_dir num_frames label1 label2 ...)
|
|
|
some/directory-1 163 1 3 5
|
|
|
some/directory-2 122 1 2
|
|
|
some/directory-3 258 2
|
|
|
|
|
|
Example of a single-label videos annotation txt file:
|
|
|
|
|
|
.. code-block:: txt
|
|
|
|
|
|
(filename label)
|
|
|
some/path/000.mp4 1
|
|
|
some/path/001.mp4 1
|
|
|
some/path/002.mp4 2
|
|
|
|
|
|
Example of a multi-label videos annotation txt file:
|
|
|
|
|
|
.. code-block:: txt
|
|
|
|
|
|
(filename label1 label2 ...)
|
|
|
some/path/000.mp4 1 3 5
|
|
|
some/path/001.mp4 1 4 8
|
|
|
some/path/002.mp4 2 4 9
|
|
|
|
|
|
Args:
|
|
|
lines (list): List of lines in 'txt' label format.
|
|
|
format (str): Data format, choices are 'rawframes' and 'videos'.
|
|
|
|
|
|
Returns:
|
|
|
list[dict]: For rawframes format, each dict has keys: frame_dir,
|
|
|
total_frames, label; for videos format, each diction has keys:
|
|
|
filename, label.
|
|
|
"""
|
|
|
lines = [x.split() for x in lines]
|
|
|
if format == 'rawframes':
|
|
|
data = [
|
|
|
dict(
|
|
|
frame_dir=line[0],
|
|
|
total_frames=int(line[1]),
|
|
|
label=[int(x) for x in line[2:]]) for line in lines
|
|
|
]
|
|
|
elif format == 'videos':
|
|
|
data = [
|
|
|
dict(filename=line[0], label=[int(x) for x in line[1:]])
|
|
|
for line in lines
|
|
|
]
|
|
|
return data
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
|
args = parse_args()
|
|
|
lines = open(args.annofile).readlines()
|
|
|
lines = [x.strip() for x in lines]
|
|
|
result = lines2dictlist(lines, args.format)
|
|
|
if args.output is None:
|
|
|
args.output = args.annofile.replace('.txt', '.json')
|
|
|
mmengine.dump(result, args.output)
|
|
|
|
