Skip to content

Commit

Permalink
Pythonclient: Add example script for changing FoV of camera
Browse files Browse the repository at this point in the history
  • Loading branch information
rajat2004 committed Apr 7, 2021
1 parent 70179ce commit 52f144b
Showing 1 changed file with 55 additions and 0 deletions.
55 changes: 55 additions & 0 deletions PythonClient/computer_vision/fov_change.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
# In settings.json first activate computer vision mode:
# https://github.com/Microsoft/AirSim/blob/master/docs/image_apis.md#computer-vision-mode

import airsim
import os
import tempfile

client = airsim.VehicleClient()
client.confirmConnection()

tmp_dir = os.path.join(tempfile.gettempdir(), "airsim_cv_mode")
print ("Saving images to %s" % tmp_dir)
try:
os.makedirs(tmp_dir)
except OSError:
if not os.path.isdir(tmp_dir):
raise

CAM_NAME = "front_center"
print(f"Camera: {CAM_NAME}")

airsim.wait_key('Press any key to get camera parameters')

cam_info = client.simGetCameraInfo(CAM_NAME)
print(cam_info)

airsim.wait_key(f'Press any key to get images, saving to {tmp_dir}')

requests = [airsim.ImageRequest(CAM_NAME, airsim.ImageType.Scene),
airsim.ImageRequest(CAM_NAME, airsim.ImageType.DepthVis)]

def save_images(responses, prefix = ""):
for i, response in enumerate(responses):
filename = os.path.join(tmp_dir, prefix + "_" + str(i))
if response.pixels_as_float:
print(f"Type {response.image_type}, size {len(response.image_data_float)}, pos {response.camera_position}")
airsim.write_pfm(os.path.normpath(filename + '.pfm'), airsim.get_pfm_array(response))
else:
print(f"Type {response.image_type}, size {len(response.image_data_uint8)}, pos {response.camera_position}")
airsim.write_file(os.path.normpath(filename + '.png'), response.image_data_uint8)


responses = client.simGetImages(requests)
save_images(responses, "old_fov")

airsim.wait_key('Press any key to change FoV and get images')

client.simSetCameraFov(CAM_NAME, 120)
responses = client.simGetImages(requests)
save_images(responses, "new_fov")

new_cam_info = client.simGetCameraInfo(CAM_NAME)
print(new_cam_info)

print(f"Old FOV: {cam_info.fov}, New FOV: {new_cam_info.fov}")

0 comments on commit 52f144b

Please sign in to comment.