opt
This commit is contained in:
parent
442b6a672d
commit
7753541fe7
@ -120,12 +120,7 @@ class Muxer(Node):
|
||||
self.playback_rf_msg = msg
|
||||
elif isinstance(msg, ImageArgMsg):
|
||||
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):
|
||||
logger.info(f'use live {msg}')
|
||||
match msg.target:
|
||||
case 'live':
|
||||
self.seq_meta_live = RfSequenceMeta.from_name(msg.name)
|
||||
|
||||
@ -131,7 +131,7 @@ class RfMat:
|
||||
seq_meta = frame.seq_meta
|
||||
else:
|
||||
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':
|
||||
m = cp.asarray(m)
|
||||
return RfMat(m, frame.meta, seq_meta)
|
||||
@ -377,6 +377,11 @@ class RfMat:
|
||||
self.m = self.m.astype(np.int64)
|
||||
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__':
|
||||
cp.zeros((1, 2, 3)) + 1
|
||||
|
||||
Loading…
Reference in New Issue
Block a user