ibaiGorordo / pyKinectAzure

Python library to run Kinect Azure DK SDK functions
MIT License
460 stars 114 forks source link

Run on GPU? #113

Open mj-raihan opened 11 months ago

mj-raihan commented 11 months ago

When I run the code, it's running on my laptop's built-in GPU, thus giving only 5 FPS. How can I run it on my laptops dedicated GPU.

laptop config: dell Inspiron 3501 i5 12th gen 24GB RAM GPU: mx330

The code is given below. def example_BodyTrackingColorCamera():

Initialize the library, if the library is not found, add the library path as argument

pykinect.initialize_libraries(track_body=True)

# Modify camera configuration
device_config = pykinect.default_configuration
device_config.color_resolution = pykinect.K4A_COLOR_RESOLUTION_1080P
device_config.depth_mode = pykinect.K4A_DEPTH_MODE_WFOV_2X2BINNED
#print(device_config)

# Start device
device = pykinect.start_device(config=device_config)

# Start body tracker
bodyTracker = pykinect.start_body_tracker()

cv2.namedWindow('Color image with skeleton',cv2.WINDOW_NORMAL)
while True:

    # Get capture
    capture = device.update()

    # Get body tracker frame
    body_frame = bodyTracker.update()

    # Get the color image
    ret, color_image = capture.get_color_image()

    if not ret:
        continue

    # Draw the skeletons into the color image
    color_skeleton = body_frame.draw_bodies(color_image, pykinect.K4A_CALIBRATION_TYPE_COLOR)

    # Overlay body segmentation on depth image
    cv2.imshow('Color image with skeleton',color_skeleton)  

    # Press q key to stop
    if cv2.waitKey(1) == ord('q'):  
        break

example_BodyTrackingColorCamera()