def _write_and_upload_video(clip_data, frames, clip_id, source, fps, gcs_path, video_local_path, clip_local_path): with open(clip_local_path, 'wb') as f: pickle.dump(clip_data, f) # Write clip to disk write_segment_to_video(frames, fname=video_local_path, fps=fps) upload_to_gcs(video_local_path, gcs_path) return clip_id, source
def test_render_videos(): env_id = "Hopper-v1" env = make_with_torque_removed(env_id) segments = segments_from_rand_rollout(env_id, make_with_torque_removed, n_desired_segments=1, clip_length_in_seconds=CLIP_LENGTH) for idx, segment in enumerate(segments): local_path = osp.join(TEST_RENDER_DIR, 'test-%s.mp4' % idx) print("Writing segment to: %s" % local_path) write_segment_to_video(segment, fname=local_path, env=env)
def test_render_videos(): env = make_with_torque_removed("Hopper-v1") collector = RandomRolloutSegmentCollector(20000, env=env) rl_teacher.agent.trpo.run_trpo_mujoco.train( num_timesteps=8000, env=env, seed=0, predictor=collector, random_rollout=True, ) segments = collector.segments tmp_media_dir = '/tmp/rl_teacher_media_test' for segment in segments: local_path = osp.join(tmp_media_dir, str(uuid.uuid4()) + '.mp4') print("Writing segment to: %s" % local_path) write_segment_to_video(segment, fname=local_path, env=env)
def _write_and_upload_video(env_id, gcs_path, local_path, segment): env = make_with_torque_removed(env_id) write_segment_to_video(segment, fname=local_path, env=env) upload_to_gcs(local_path, gcs_path)
def _write_and_upload_video( local_path, segment): # logger.info('write video !!!'+local_path) write_segment_to_video(segment, fname=local_path)
def _write_and_upload_video(env_id, gcs_path, local_path, segment): os.mkdir('/tmp/233') env = make_with_torque_removed(env_id) write_segment_to_video(segment, fname=local_path, env=env)