mindsight/requirements.txt at main · kylen-d/mindsight · GitHub

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
# MindSight core dependencies
# Install with: pip install -r requirements.txt
# For GPU acceleration, install PyTorch separately first:
#   https://pytorch.org/get-started/locally/

# Deep learning
torch>=2.0
torchvision>=0.15
onnxruntime>=1.16

# Object detection
ultralytics>=8.0
clip @ git+https://github.com/ultralytics/CLIP.git@88ade288431a46233f1556d1e141901b3ef0a36b

# Gaze estimation
uniface>=1.0                    # RetinaFace face detector
timm>=0.3.2                     # PyTorch Image Models (required by UniGaze)
# unigaze                       # UniGaze backend — install separately if needed
                                # (non-commercial license, pins timm==0.3.2)

# Computer vision
opencv-python>=4.8
opencv-contrib-python>=4.8
Pillow>=10.0

# GUI
PyQt6>=6.5

# Data & analysis
numpy>=1.24
matplotlib>=3.7
pandas>=2.0

# Utilities
PyYAML>=6.0
tqdm>=4.65