Updated requirements and README

README.md

@@ -17,6 +17,26 @@ Official PyTorch Implementation of the <a href="https://github.com/CASIA-IVA-Lab
 The **Fast Segment Anything Model(FastSAM)** is a CNN Segment Anything Model trained by only 2% of the SA-1B dataset published by SAM authors. The FastSAM achieve a comparable performance with
 the SAM method at **50× higher run-time speed**.
 
+## Local Setup (Anaconda Environment Recommended)
+
+* Create a new conda environment
+
+```
+conda create -n fastsam python=3.11
+```
+
+* Install PyTorch 2.5.0 with CUDA 12.4
+
+```
+conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia
+```
+
+* Install rest of the requirements
+
+```
+pip install -r requirements.txt
+```
+
 
 ## License
 

requirements.txt

@@ -1,20 +1,115 @@
-# Base-----------------------------------
-matplotlib==3.2.2
-numpy
-opencv-python
-
+aiofiles==23.2.1
+aiohappyeyeballs==2.4.3
+aiohttp==3.10.10
+aiosignal==1.3.1
+altair==5.4.1
+annotated-types==0.7.0
+anyio==4.6.2.post1
+async-timeout==4.0.3
+attrs==24.2.0
+Authlib==1.3.2
+certifi==2024.8.30
+cffi==1.17.1
+charset-normalizer==3.4.0
+click==8.0.4
 git+https://github.com/openai/CLIP.git
-# Pillow>=7.1.2
-# PyYAML>=5.3.1
-# requests>=2.23.0
-# scipy>=1.4.1
-# torch
-# torchvision
-# tqdm>=4.64.0
-
-# pandas>=1.1.4
-# seaborn>=0.11.0
-
-# Ultralytics-----------------------------------
+cryptography==43.0.3
+cycler==0.12.1
+datasets==3.0.2
+dill==0.3.8
+exceptiongroup==1.2.2
+fastapi==0.112.4
+ffmpy==0.4.0
+filelock==3.16.1
+frozenlist==1.5.0
+fsspec==2024.9.0
+ftfy==6.3.0
+gradio==4.36.1
+gradio_client==1.0.1
+h11==0.14.0
+hf_transfer==0.1.8
+httpcore==1.0.6
+httpx==0.27.2
+huggingface-hub==0.26.1
+idna==3.10
+importlib_resources==6.4.5
+itsdangerous==2.2.0
+Jinja2==3.1.4
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+kiwisolver==1.4.7
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.2.2
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.1.0
+multiprocess==0.70.16
+narwhals==1.10.0
+networkx==3.4.2
+numpy==2.1.2
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
+opencv-python==4.10.0.84
+orjson==3.10.10
+packaging==24.1
+pandas==2.2.3
+pillow==10.4.0
+pip==22.3.1
+propcache==0.2.0
+protobuf==3.20.3
+psutil==5.9.8
+pyarrow==17.0.0
+pycparser==2.22
+pydantic==2.9.2
+pydantic_core==2.23.4
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.2.0
+python-dateutil==2.9.0.post0
+python-multipart==0.0.12
+pytz==2024.2
+PyYAML==6.0.2
+referencing==0.35.1
+regex==2024.9.11
+requests==2.32.3
+rich==13.9.3
+rpds-py==0.20.0
+ruff==0.7.0
+scipy==1.14.1
+seaborn==0.12.2
+semantic-version==2.10.0
+setuptools==65.5.1
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+spaces==0.30.4
+starlette==0.38.6
+sympy==1.13.1
+tomlkit==0.12.0
+torch==2.5.0
+torchvision==0.20.0
+tqdm==4.66.5
+triton==3.1.0
+typer==0.12.5
+typing_extensions==4.12.2
+tzdata==2024.2
 ultralytics==8.0.121
+urllib3==2.2.3
+uvicorn==0.32.0
+wcwidth==0.2.13
+websockets==11.0.3
+wheel==0.44.0
+xxhash==3.5.0
+yarl==1.16.0