Skip to content

Commit 6c13100

Browse files
committed
small fix in concatenator script
1 parent 2329ed1 commit 6c13100

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dnn/torch/osce/scripts/concatenator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
parser.add_argument("--db_max", type=float, help="max DB for random normalization", default=0)
2020
parser.add_argument("--db_min", type=float, help="min DB for random normalization", default=0)
2121
parser.add_argument("--verbose", action="store_true")
22+
parser.add_argument("--seed", type=int, help="random seed", default=0)
2223

2324
def read_filelist(basedir, filelist):
2425
with open(filelist, "r") as f:
@@ -74,12 +75,14 @@ def concatenate(filelist : str, output : str, target_fs: int, normalize=True, db
7475

7576
f.write(x1.astype(np.int16).tobytes())
7677

77-
overlap_mem = x1[-overlap_size]
78+
overlap_mem = x1[-overlap_size:]
7879

7980

8081
if __name__ == "__main__":
8182
args = parser.parse_args()
8283

84+
np.random.seed(args.seed)
85+
8386
filelist = read_filelist(args.basedir, args.filelist)
8487

8588
concatenate(filelist, args.output, args.target_fs, normalize=args.normalize, db_min=args.db_min, db_max=args.db_max, verbose=args.verbose)

0 commit comments

Comments
 (0)