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?