YanCotta commited on
Commit
971d800
·
verified ·
1 Parent(s): 7aac01d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -4
README.md CHANGED
@@ -1,10 +1,48 @@
1
  ---
 
 
 
 
2
  tags:
3
  - object-detection
4
  - yolo
 
5
  - framenet
6
- library_name: ultralytics
7
  ---
8
- # ReINVenTA: YOLOv8 Object Detector
9
- Modelo de detecção de objetos treinado no dataset Flickr30k Entities (Estágio 1).
10
- Parte do projeto **ReINVenTA** do laboratório **FrameNetBrasil**.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ library_name: ultralytics
3
+ license: agpl-3.0
4
+ language: pt
5
+ pipeline_tag: object-detection
6
  tags:
7
  - object-detection
8
  - yolo
9
+ - yolov8
10
  - framenet
11
+ - pytorch
12
  ---
13
+
14
+ # 👁️ ReINVenTA: YOLOv8 Object Detector (Stage 1)
15
+
16
+ > **Detector de Elementos de Frame (Fine-Tuned YOLOv8-Medium)**
17
+
18
+ Este modelo representa o **Estágio 1 (Percepção Simbólica)** do pipeline **ReINVenTA**. Ele atua como os "olhos" do sistema, transformando pixels brutos em uma lista estruturada de entidades presentes na cena.
19
+
20
+ ## 🚀 Sobre o Modelo
21
+ Treinado no dataset **Flickr30k Entities**, este modelo foi adaptado para detectar **635 tipos de Elementos de Frame** (ex: `People`, `Clothing`, `Animals`, `Vehicles`, `Tools`).
22
+
23
+ - **Base Model:** YOLOv8m (Medium)
24
+ - **Dataset:** Flickr30k Entities (Bounding Boxes)
25
+ - **Função:** Detecção de objetos para construção de vetor simbólico (Neuro-Symbolic Fusion).
26
+
27
+ ## 💻 Como Usar
28
+
29
+ Para utilizar este modelo, você precisa da biblioteca `ultralytics`.
30
+
31
+ ```bash
32
+ pip install ultralytics huggingface_hub
33
+ from ultralytics import YOLO
34
+ from huggingface_hub import hf_hub_download
35
+
36
+ # 1. Baixar o modelo do Hub
37
+ model_path = hf_hub_download(repo_id="FrameNetBrasil/reinventa-yolo-detection", filename="best.pt")
38
+
39
+ # 2. Carregar com Ultralytics
40
+ model = YOLO(model_path)
41
+
42
+ # 3. Inferência
43
+ results = model("[https://ultralytics.com/images/bus.jpg](https://ultralytics.com/images/bus.jpg)") # Aceita URL ou caminho local
44
+
45
+ # 4. Resultados
46
+ for result in results:
47
+ result.show() # Mostra a imagem com caixas
48
+ print(result.names) # Lista de classes detectadas