cog.yaml

build:
  python_version: "3.8"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "cachetools==4.1.0"
    - "chardet==3.0.4"
    - "future==0.18.2"
    - "fvcore==0.1.dev200506"
    - "idna==2.9"
    - "importlib-metadata==1.6.0"
    - "jsonpatch==1.25"
    - "jsonpointer==2.0"
    - "markdown==3.2.2"
    - "mock==4.0.2"
    - "opencv-python==4.3.0.38"
    - "portalocker==1.7.0"
    - "pyasn1==0.4.8"
    - "pyasn1-modules==0.2.8"
    - "pydot==1.4.1"
    - "requests==2.23.0"
    - "requests-oauthlib==1.3.0"
    - "rsa==4.0"
    - "tabulate==0.8.7"
    - "termcolor==1.1.0"
    - "urllib3==1.25.8"
    - "visdom==0.1.8.9"
    - "websocket-client==0.57.0"
    - "yacs==0.1.7"
    - "zipp==3.1.0"
    - "cython==0.29.22"
    - "pyyaml==5.1"
    - "dominate==2.4.0"
    - "detectron2==0.1.2"
    - "torch==1.5.0"
    - "torchvision==0.6.0"
    - "pycocotools==2.0.2"
    - "ipython==7.21.0"
    - "scikit-image==0.18.1"
  python_extra_index_urls:
    - "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"
  python_find_links:
    - "https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/index.html"
predict: "predict.py:InstColorizationPredictor"