def cutter_test():
# Define a stream reader
reader = stream.StreamReader(
waveFile=wavPath,
chunkSize=480,
simulate=False,
)
# Define a cutter
# if batch size is 1, the output will be a vector (one frame)
# otherwise, the output will be a matrix (a batch frames)
cutter = stream.ElementFrameCutter(
batchSize=1,
width=400,
shift=160,
)
# Start
reader.start()
cutter.start(inPIPE=reader.outPIPE)
cutter.wait()
print(cutter.outPIPE.size())
pac = cutter.outPIPE.get()
print(pac.keys())
print(pac[pac.mainKey])
def detector_test():
reader = stream.StreamReader(
waveFile=wavPath,
chunkSize=480,
simulate=False,
)
cutter = stream.ElementFrameCutter(
batchSize=1,
width=400,
shift=160,
)
detector = stream.VectorVADetector(batchSize=50, vadFunc=lambda x: True)
chain = base.Chain()
chain.add(reader)
chain.add(cutter)
chain.add(detector)
chain.start()
chain.wait()
print(chain.outPIPE.size())
def subsetter_test():
reader = stream.StreamReader(
waveFile=wavPath,
chunkSize=480,
simulate=False,
)
cutter = stream.ElementFrameCutter(
batchSize=50,
width=400,
shift=160,
)
subsetter = stream.MatrixSubsetter(nChunk=2, )
chain = base.Chain()
chain.add(reader)
chain.add(cutter)
chain.add(subsetter)
chain.start()
chain.wait()
print(chain.outPIPE.size())
def batcher_test():
reader = stream.StreamReader(
waveFile=wavPath,
chunkSize=480,
simulate=False,
)
cutter = stream.ElementFrameCutter(
batchSize=1,
width=400,
shift=160,
)
batcher = stream.VectorBatcher(center=50, )
chain = base.Chain()
chain.add(reader)
chain.add(cutter)
chain.add(batcher)
chain.start()
chain.wait()
print(chain.outPIPE.size())
def stream_recorder_cutter_test():
recorder = stream.StreamRecorder(oKey="stream")
cutter = stream.ElementFrameCutter(batchSize=50,
width=400,
shift=160,
oKey="frames")
cutter.link(inPIPE=recorder.outPIPE, iKey="stream")
recorder.start()
cutter.start()
time.sleep(2)
recorder.stop()
cutter.wait()
print(cutter.outPIPE.size())
print(cutter.outPIPE.get().keys())
def stream_recorder_cutter_chain_test():
recorder = stream.StreamRecorder(oKey="stream")
cutter = stream.ElementFrameCutter(batchSize=50,
width=400,
shift=160,
oKey="frames")
chain = base.Chain()
chain.add(node=recorder)
chain.add(cutter, iKey="stream")
chain.start()
time.sleep(2)
chain.stop()
chain.wait()
print("size:", chain.outPIPE.size())
#stream_recorder_cutter_chain_test()