environment.yml

name: cropmask
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  # from solaris env-gpu file
  - python>=3.6
  - pip>=19.0.3
  - affine>=2.3.0
  - albumentations=0.4.3
  - fiona>=1.7.13
  - gdal>=3.0.2
  - geopandas>=0.6.2
  - matplotlib>=3.1.2
  - networkx>=2.4
  - numpy>=1.17.3
  - opencv>=4.1
  - pandas>=0.25.3
  - pyproj>=2.1
  - pytorch=1.3.1
  - pyyaml=5.2
  - rasterio>=1.0.23
  - requests=2.22.0
  - rio-cogeo>=1.1.6
  - rtree>=0.9.3
  - scikit-image>=0.16.2
  - scipy>=1.3.2
  - shapely>=1.6.4
  - torchvision>=0.4.2
  - tqdm>=4.40.0
  - urllib3>=1.25.7
  - tensorflow-gpu=1.14
  - tensorboard

  # scientific python
  - pyqt
  - xlrd # for excel sheets in pandas
  - xarray
  - dask

  # matterport Mask_RCNN requirements
  - Pillow==6.1 # pillow 7 breaks torchvision
  - cython
  - h5py
  - IPython[all]
  - keras-gpu

  # Geo stuff
  - rasterstats
  - rioxarray
  - geocube
  - geopy
  - cartopy
  - contextily
  - pysal
  - pyproj
  - folium
  - kealib
  - geojson

  # Other stuff
  - yaml
  - jupyterlab

  - pip:
      - imgaug # for mrcnn
      - us
      - graphviz
      #detectron2 stuff
      - git+git://github.com/waspinator/coco.git@2.1.0
      - git+git://github.com/waspinator/pycococreator.git@0.2.0
      - git+https://github.com/facebookresearch/fvcore
      - flit # for installing cropmask in dev env