TowardsAurora

I also meet the problem. But when I change my dataset, the problem breake down. you can use labelme to label your dataset and use the under code to reshape...