prostochel097 commited on
Commit
1a93343
·
verified ·
1 Parent(s): 14414c2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +69 -3
README.md CHANGED
@@ -1,3 +1,69 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - depth-estimation
5
+ - robotics
6
+ - computer-vision
7
+ - pytorch
8
+ - safetensors
9
+ datasets:
10
+ - custom-pybullet-synthetic
11
+ metrics:
12
+ - mse
13
+ ---
14
+
15
+ # AlphaDepth v1.0 👁️
16
+
17
+ **AlphaDepth** — это легковесная нейросеть для оценки глубины (Depth Estimation), обученная в симуляции PyBullet. Модель принимает RGB-изображение и предсказывает карту глубины (Depth Map), позволяя роботам "видеть" расстояние до объектов с помощью одной камеры.
18
+
19
+ Входит в серию моделей **Alpha**.
20
+
21
+ ## 📊 Характеристики
22
+
23
+ - **Архитектура:** Custom U-Net с BatchNorm
24
+ - **Вход:** RGB Image (128x128)
25
+ - **Выход:** Depth Map (Normalized 0..1)
26
+ - **Вес:** ~1.5 MB (FP32)
27
+ - **Формат:** SafeTensors
28
+
29
+ ## 🚀 Использование
30
+
31
+ Вам понадобится `safetensors` и файл `model.py` из этого репозитория.
32
+
33
+ ```python
34
+ import torch
35
+ from model import AlphaUNet
36
+ from PIL import Image
37
+ from torchvision import transforms
38
+
39
+ # 1. Загрузка модели
40
+ model = AlphaUNet.from_pretrained(".") # Укажите путь к папке
41
+ model.eval()
42
+
43
+ # 2. Подготовка изображения
44
+ img = Image.open("robot_view.jpg").resize((128, 128))
45
+ transform = transforms.Compose([
46
+ transforms.ToTensor(), # 0..255 -> 0..1
47
+ ])
48
+ input_tensor = transform(img).unsqueeze(0)
49
+
50
+ # 3. Инференс
51
+ with torch.no_grad():
52
+ depth_map = model(input_tensor)
53
+
54
+ # depth_map теперь содержит расстояния!
55
+ ```
56
+ ## 🛠 Обучение
57
+
58
+ Модель обучалась на **синтетическом датасете**, сгенерированном в движке PyBullet.
59
+ - **Эпохи:** 20
60
+ - **Loss:** 0.0016 (MSE)
61
+ - **Оптимизатор:** Adam (lr=0.001)
62
+
63
+ ## 📦 Файлы
64
+
65
+ - `model.safetensors`: Веса модели (SafeTensors формат).
66
+ - `config.json`: Конфигурация архитектуры.
67
+ - `model.py`: Исходный код класса нейросети.
68
+
69
+ *Created by prostochel097 for the Alpha Series.*