driesverachtert/basic_shapes_object_detection_model

Model that is fine tuned to detect some basic geometric shapes

language	license	pretty_name	tags	task_categories
en	apache-2.0	Basic Shapes Object Detection Model	object-detection simple example basic-geometric-shapes	object-detection

Basic Shapes Object Detection Model

Description

This is a model based on facebook/detr-resnet-50 which is fine-tuned to detect very basic shapes like circles, squares and triangles.

It's trained on the dataset at https://huggingface.co/datasets/driesverachtert/basic_shapes_object_detection . The code which was used to create that dataset can be found at https://github.com/DriesVerachtert/basic_shapes_object_detection_dataset

This model itself can be downloaded from the HuggingFace Hub at https://huggingface.co/driesverachtert/basic_shapes_object_detection . The code which was used to train the model can be found at https://github.com/DriesVerachtert/basic_shapes_object_detection_model

License

This model is released under Apache 2.0.

Useful links

• https://github.com/facebookresearch/detr
• https://huggingface.co/facebook/detr-resnet-50
• https://huggingface.co/docs/transformers/main/en/tasks/object_detection
• https://huggingface.co/docs/transformers/main/en/model_doc/detr
• https://github.com/Rishit-dagli/CPPE-Dataset

Contact

Dries Verachtert - [email protected] https://www.linkedin.com/in/dries/