32 lines
644 B
YAML
32 lines
644 B
YAML
name: seg_smp
|
|
channels:
|
|
- conda-forge
|
|
- defaults
|
|
dependencies:
|
|
- python=3.11
|
|
- pip
|
|
- pip:
|
|
- --extra-index-url https://download.pytorch.org/whl/cu124
|
|
- fastapi>=0.110
|
|
- uvicorn[standard]>=0.27
|
|
- pydantic>=2
|
|
- python-multipart>=0.0.9
|
|
- pytest>=8
|
|
- torch==2.6.0
|
|
- torchvision==0.21.0
|
|
- opencv-python<4.12
|
|
- numpy<2
|
|
- albumentations
|
|
- segmentation-models-pytorch
|
|
- ultralytics
|
|
- mmengine
|
|
- mmsegmentation==1.2.2
|
|
- mmcv-lite
|
|
- matplotlib
|
|
- pandas
|
|
- scikit-learn
|
|
- scipy
|
|
- seaborn
|
|
- tqdm
|
|
- tensorboard
|