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()