[U] Backup unfinished changes

This commit is contained in:
Azalea (on HyDEV-Daisy)
2022-10-01 13:41:07 -04:00
parent b452d77cac
commit c39ce766af
5 changed files with 255 additions and 37 deletions
+47 -2
View File
@@ -38,8 +38,53 @@ class Result(NamedTuple):
file: str
def segment(file) -> list[ResultFrame]:
return [ResultFrame(*s) for s in seg(file)]
class BatchResults(NamedTuple):
results: list[Result]
time_full: float
time_avg: float
successes: int
messages: list[tuple[str, int]]
def _process_helper(args: tuple[Segmenter, str]) -> Result:
seg, f = args
lseg = seg(f)
return Result([ResultFrame(*s) for s in lseg], f)
def process(self: Segmenter, inp: list[str], tmpdir=None, verbose=False, skip_if_exist=False,
nbtry=1, try_delay=2.) -> list[Result]:
from tqdm.contrib.concurrent import process_map
import tqdm
results = []
for i in tqdm.tqdm(inp):
results.append(_process_helper((self, i)))
return results
# return process_map(_process_helper, [(self, i) for i in inp], max_workers=2)
#
# t_batch_start = time.time()
#
# results: list[Result] = []
# lmsg = []
# fg = featGenerator(inp.copy(), inp.copy(), tmpdir, self.ffmpeg, skip_if_exist, nbtry, try_delay)
# i = 0
# for feats, msg in fg:
# lmsg += msg
# i += len(msg)
# if verbose:
# print('%d/%d' % (i, len(inp)), msg)
# if feats is None:
# break
# mspec, loge, diff_len = feats
# lseg = self.segment_feats(mspec, loge, diff_len, 0)
# results.append()
#
# t_batch_dur = time.time() - t_batch_start
# nb_processed = len([e for e in lmsg if e[1] == 0])
# avg = t_batch_dur / nb_processed if nb_processed else -1
# return BatchResults(results, t_batch_dur, avg, nb_processed, lmsg)
def to_wav(file: str, callback: Callable, start_sec: float = 0, stop_sec: float = 0):