process_frame_p3.py

from scheduling.misc import *
from scheduling.TaskEntity import *


# read the input bounding box data from file
box_info = read_json_file('../dataset/waymo_ground_truth_flat.json')


def process_frame(frame):
    """Process frame for scheduling.

    Process a image frame to obtain cluster boxes and corresponding scheduling parameters
    for scheduling. 

    Student's code here.

    Args:
        param1: The image frame to be processed. 

    Returns:
        A list of tasks with each task containing image_path and other necessary information. 
    """

    cluster_boxes_data = get_bbox_info(frame, box_info)
    cluster_boxes_data.sort(key=lambda x : x[4])
    # student's code here
    task = []
    for i in range(len(cluster_boxes_data)):
        task.append(TaskEntity(frame.path, coord = cluster_boxes_data[i][:4], priority = i,depth = cluster_boxes_data[i][4]))
    return task