harshinde commited on
Commit
6eb4e0f
·
verified ·
1 Parent(s): bbcbc4e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +31 -3
requirements.txt CHANGED
@@ -1,3 +1,31 @@
1
- altair
2
- pandas
3
- streamlit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core dependencies for Landslide Detection Project
2
+ streamlit>=1.28.0
3
+ torch>=2.0.0
4
+ torchvision>=0.15.0
5
+ numpy>=1.24.0
6
+ matplotlib>=3.7.0
7
+ h5py>=3.8.0
8
+ PyYAML>=6.0
9
+
10
+ # Deep Learning and Computer Vision
11
+ segmentation-models-pytorch>=0.3.0
12
+ torchmetrics>=1.0.0
13
+ pytorch-lightning>=2.0.0
14
+
15
+ # Transformers for SegFormer model
16
+ transformers>=4.30.0
17
+
18
+ # Logging and Experiment Tracking
19
+ wandb>=0.15.0
20
+
21
+ # Additional utilities
22
+ Pillow>=9.5.0
23
+ scikit-image>=0.20.0
24
+ opencv-python>=4.8.0
25
+
26
+ # Kaggle API for model downloads
27
+ kaggle>=1.5.12
28
+
29
+ # Optional: For better performance
30
+ # torch-scatter>=2.1.0 # Uncomment if needed for specific models
31
+ # torch-sparse>=0.6.0 # Uncomment if needed for specific models