Is there any specific model and dataset to train a particular shape?. For example Object detection of rectangle.