| language | license | pretty_name | tags | task_categories | ||||||
|---|---|---|---|---|---|---|---|---|---|---|
|
apache-2.0 |
Basic Shapes Object Detection Model |
|
|
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
This model is released under Apache 2.0.
- 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
Dries Verachtert - dries.verachtert@dries.eu https://www.linkedin.com/in/dries/