TyurinYuriRost commited on
Commit
0032bc8
·
verified ·
1 Parent(s): 8dda912

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -0
README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ task: object-detection
3
+ library: pytorch
4
+ tags:
5
+ - object detection
6
+ - yolo
7
+ - floor plan
8
+ - image classification
9
+ dataset: imagenet # Если вы использовали imagenet как основу или похожий
10
+ trained_dataset: custom floor plan dataset
11
+ ---
12
+ **Автор:** TyurinYuriRost
13
+
14
+ **Модель:** Floor Plan YOLO
15
+
16
+ **Описание:**
17
+ Это модель YOLO, обученная для распознавания и локализации объектов на планах этажей. Она может идентифицировать такие элементы, как двери, окна, стены, мебель и т.д.
18
+
19
+ **Обучение:**
20
+ * **Задача:** Обнаружение объектов (Object Detection)
21
+ * **Библиотека:** PyTorch
22
+ * **Архитектура:** YOLOv8 (или какая у вас версия)
23
+ * **Обучающий набор данных:** Пользовательский набор данных планов этажей (или упомяните, если использовали какой-то известный датасет как основу)
24
+ * **Параметры обучения:** (Например, количество эпох, размер батча, оптимизатор, learning rate)
25
+
26
+ **Использование:**
27
+
28
+ **Установка:**
29
+ ```bash
30
+ pip install huggingface_hub
31
+ (Укажите другие необходимые библиотеки, если есть)
32
+ Загрузка и использование модели:
33
+ code
34
+ Python
35
+ from huggingface_hub import HfApi, hf_hub_download
36
+ from PIL import Image
37
+ import torch
38
+
39
+ # Загрузка файла модели (пример)
40
+ model_path = hf_hub_download(repo_id="TyurinYuriRost/super-cool-model", filename="floor_plan_yolo.pt")
41
+
42
+ # Пример загрузки весов модели (если вы используете PyTorch)
43
+ # model = torch.load(model_path)
44
+ # model.eval()
45
+
46
+ # Пример использования (замените на ваш реальный код инференса)
47
+ # image = Image.open("path/to/your/floor_plan.jpg")
48
+ # results = model(image)
49
+ # print(results)
50
+ Демонстрация:
51
+
52
+ Репозиторий
53
+ Hugging Face: https://huggingface.co/TyurinYuriRost/super-cool-model