File size: 1,277 Bytes
4095301 |
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 39 40 41 42 |
import os
import os.path as osp
import argparse
import tqdm
SAMPLE_RATE = 16000
def main(args):
print(args)
with open(args.input, 'r') as fr:
lines = fr.readlines()
total_frames = 0
total_seconds = 0.
max_frames = (0, 0)
for i, l in tqdm.tqdm(enumerate(lines[1:]), total=len(lines[1:])):
vid, frames_str = l.strip().split('\t')
# print(frames_str)
frames = int(frames_str)
if frames > max_frames[1]:
max_frames = (i, frames)
total_frames += frames
total_seconds += frames / SAMPLE_RATE
total_lines = len(lines[1:])
print(f"Total frames: {total_frames}")
print(f"Total audios: {total_lines}")
print("==================================")
print(f"Average frames: {total_frames / total_lines}")
print(f"Max frames: {max_frames}")
print(f"Total seconds: {total_seconds}")
print(f"{total_seconds // 3600} hours, {total_seconds % 3600 // 60} minutes and {total_seconds % 60} seconds")
print(f"Average seconds: {total_seconds / total_lines}")
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument(
"--input",
default="",
help="a sample arg",
)
args = parser.parse_args()
main(args) |