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 test_stream_reader():
# Define a stream reader
reader = stream.StreamReader(
waveFile=wavPath,
chunkSize=480,
simulate=False,
#oKey="data",
)
reader.start()
reader.wait()
# Get the output PIPE and packet
print(reader.outPIPE.size())
pac = reader.outPIPE.get()
print(pac.mainKey)
print(pac.keys())
print(pac[pac.mainKey])
def test_stream_reader_vad():
# Define a stream reader
# The webrtc VAD is used
reader = stream.StreamReader(
waveFile=wavPath,
chunkSize=480,
simulate=False,
vaDetector=stream.WebrtcVADetector(),
)
reader.start()
reader.wait()
# Get the output PIPE and packet
print(reader.outPIPE.size())
pac = reader.outPIPE.get()
print(pac.mainKey)
print(pac.keys())
print(pac[pac.mainKey])