I have a dataset of 96 traffic signs (9000 total photos splitted in 96 categories), I wrote this model which loads the classnum from a json, modify the images to have a simi