opt
This commit is contained in:
parent
442b6a672d
commit
7753541fe7
@ -120,12 +120,7 @@ class Muxer(Node):
|
|||||||
self.playback_rf_msg = msg
|
self.playback_rf_msg = msg
|
||||||
elif isinstance(msg, ImageArgMsg):
|
elif isinstance(msg, ImageArgMsg):
|
||||||
self.arg = msg
|
self.arg = msg
|
||||||
elif isinstance(msg, SetSeqMetaMsg):
|
|
||||||
match msg.target:
|
|
||||||
case 'playback':
|
|
||||||
self.seq_meta_playback = RfSequenceMeta.from_name(msg.name)
|
|
||||||
elif isinstance(msg, SeqMetaMsg):
|
elif isinstance(msg, SeqMetaMsg):
|
||||||
logger.info(f'use live {msg}')
|
|
||||||
match msg.target:
|
match msg.target:
|
||||||
case 'live':
|
case 'live':
|
||||||
self.seq_meta_live = RfSequenceMeta.from_name(msg.name)
|
self.seq_meta_live = RfSequenceMeta.from_name(msg.name)
|
||||||
|
|||||||
@ -131,7 +131,7 @@ class RfMat:
|
|||||||
seq_meta = frame.seq_meta
|
seq_meta = frame.seq_meta
|
||||||
else:
|
else:
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
m = np.frombuffer(frame.__bytes__(), dtype=np.int16).reshape(seq_meta.shape)
|
m = np.frombuffer(frame.__bytes__(), dtype=np.int16).reshape(seq_meta.shape).copy()
|
||||||
if device == 'gpu':
|
if device == 'gpu':
|
||||||
m = cp.asarray(m)
|
m = cp.asarray(m)
|
||||||
return RfMat(m, frame.meta, seq_meta)
|
return RfMat(m, frame.meta, seq_meta)
|
||||||
@ -377,6 +377,11 @@ class RfMat:
|
|||||||
self.m = self.m.astype(np.int64)
|
self.m = self.m.astype(np.int64)
|
||||||
return self
|
return self
|
||||||
|
|
||||||
|
def jupyter(self):
|
||||||
|
from matplotlib import pyplot as plt
|
||||||
|
plt.figure(figsize=(40, 20))
|
||||||
|
plt.imshow(self.m, cmap='grey')
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
cp.zeros((1, 2, 3)) + 1
|
cp.zeros((1, 2, 3)) + 1
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user