Skip to content

Commit 10f9af6

Browse files
committed
update readme
1 parent 5d695f9 commit 10f9af6

File tree

3 files changed

+19
-0
lines changed

3 files changed

+19
-0
lines changed

README.md

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,19 @@ python3 eval_multipro.py --gpus GPUS --cfg config/ade20k-resnet50dilated-ppm_dee
228228
python3 eval_multipro.py --gpus GPUS --cfg config/ade20k-resnet101-upernet.yaml
229229
```
230230

231+
## Integration with other projects
232+
This library can be installed via `pip` to easily integrate with another codebase
233+
```bash
234+
pip install git+https://github.com/CSAILVision/semantic-segmentation-pytorch.git@master
235+
```
236+
237+
Now this library can easily be consumed programmatically. For example
238+
```python
239+
from mit_semseg.config import cfg
240+
from mit_semseg.dataset import TestDataset
241+
from mit_semseg.models import ModelBuilder, SegmentationModule
242+
```
243+
231244
## Reference
232245

233246
If you find the code or pre-trained models useful, please cite the following papers:

mit_semseg/__init__.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
"""
2+
MIT CSAIL Semantic Segmentation
3+
"""
4+
5+
__version__ = '1.0.0'

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
'torchvision',
2424
'opencv-python',
2525
'yacs',
26+
'scipy',
2627
'tqdm'
2728
]
2829
)

0 commit comments

Comments
 (0)