def test_video(self):
try:
import moviepy # noqa F401
except ImportError:
return
self.assertTrue(compare_proto(summary.video('dummy', np.random.rand(4, 3, 1, 8, 8).astype(np.float32)), self))
summary.video('dummy', np.random.rand(16, 48, 1, 28, 28).astype(np.float32))
summary.video('dummy', np.random.rand(20, 7, 1, 8, 8).astype(np.float32))
def test_video(self):
try:
import moviepy # noqa: F401
except ImportError:
return
self.assertTrue(compare_proto(summary.video('dummy', tensor_N(shape=(4, 3, 1, 8, 8))), self))
summary.video('dummy', np.random.rand(16, 48, 1, 28, 28))
summary.video('dummy', np.random.rand(20, 7, 1, 8, 8))
def add_video(self,
tag,
vid_tensor,
global_step=None,
fps=4,
walltime=None):
torch._C._log_api_usage_once("tensorboard.logging.add_video")
self._get_file_writer().add_summary(video(tag, vid_tensor, fps),
global_step, walltime)
