Error custom eager few-shot code

I have adapted the eager few-shot code to detect two objects viz traffic light detection. The purpose is to be able to detect red lights and green lights. I have been able to process the data into what I assume to be the right format, however I run into a ValueError during training. I have attached the notebook Custom Obejct Detection.ipynb (872.5 KB) with the error for your perusal.