File size: 1,347 Bytes
8e711e6 2267709 8e711e6 2267709 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
---
language:
- en
license: apache-2.0
pretty_name: Basic Shapes Object Detection Model
tags:
- object-detection
- simple
- example
- basic-geometric-shapes
task_categories:
- 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 - dries.verachtert@dries.eu
https://www.linkedin.com/in/dries/
|