driesverachtert commited on
Commit
2267709
1 Parent(s): 4b12e2b

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +39 -1
README.md CHANGED
@@ -1,3 +1,41 @@
1
  ---
2
- {}
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language:
3
+ - en
4
+ license: apache-2.0
5
+ pretty_name: Basic Shapes Object Detection Model
6
+ tags:
7
+ - object-detection
8
+ - simple
9
+ - example
10
+ - basic-geometric-shapes
11
+ task_categories:
12
+ - object-detection
13
  ---
14
+
15
+ # Basic Shapes Object Detection Model
16
+
17
+ ## Description
18
+
19
+ This is a model based on facebook/detr-resnet-50 which is fine-tuned to detect very basic shapes like circles, squares and triangles.
20
+
21
+ 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
22
+
23
+ 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
24
+
25
+ ## License
26
+
27
+ This model is released under Apache 2.0.
28
+
29
+
30
+ ## Useful links
31
+
32
+ * https://github.com/facebookresearch/detr
33
+ * https://huggingface.co/facebook/detr-resnet-50
34
+ * https://huggingface.co/docs/transformers/main/en/tasks/object_detection
35
+ * https://huggingface.co/docs/transformers/main/en/model_doc/detr
36
+ * https://github.com/Rishit-dagli/CPPE-Dataset
37
+
38
+ ## Contact
39
+
40
+ Dries Verachtert - [email protected]
41
+ https://www.linkedin.com/in/dries/