space
This commit is contained in:
@@ -199,10 +199,10 @@ def prepare_audio_frame(temp_audio_frame : AudioFrame) -> AudioFrame:
|
|||||||
|
|
||||||
def prepare_close_frame(close_vision_frame : VisionFrame) -> VisionFrame:
|
def prepare_close_frame(close_vision_frame : VisionFrame) -> VisionFrame:
|
||||||
model_size = get_model_options().get('size')
|
model_size = get_model_options().get('size')
|
||||||
close_vision_frame = numpy.expand_dims(close_vision_frame, axis=0)
|
close_vision_frame = numpy.expand_dims(close_vision_frame, axis = 0)
|
||||||
prepare_vision_frame = close_vision_frame.copy()
|
prepare_vision_frame = close_vision_frame.copy()
|
||||||
prepare_vision_frame[:, model_size[0] // 2:] = 0
|
prepare_vision_frame[:, model_size[0] // 2:] = 0
|
||||||
close_vision_frame = numpy.concatenate((prepare_vision_frame, close_vision_frame), axis=3)
|
close_vision_frame = numpy.concatenate((prepare_vision_frame, close_vision_frame), axis = 3)
|
||||||
close_vision_frame = close_vision_frame.transpose(0, 3, 1, 2).astype('float32') / 255.0
|
close_vision_frame = close_vision_frame.transpose(0, 3, 1, 2).astype('float32') / 255.0
|
||||||
return close_vision_frame
|
return close_vision_frame
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user