About the agent class in the Host code? NameError: name 'Agent' is not defined

[matlinsas]

Hi ,
I am trying to replay this project in the pynq z2, and I can find following class Agent is not defined. Could you please help to share the code of the class agent?

agent = Agent(team)
NameError: name 'Agent' is not defined

agent.save_results_xml(boxes)
agent.write(tbatch, total_num_img, team)

Thanks,
Matlinsas

[matlinsas]

And the function of the get_image_batch is not defined yet..
Please help to check it.

[toebees]

Having same issue here. Have you found a fix?