build: | |
gpu: true | |
python_version: "3.8" | |
system_packages: | |
- "libgl1-mesa-glx" | |
- "libglib2.0-0" | |
python_packages: | |
- "torchvision==0.9.0" | |
- "torch==1.8.0" | |
- "numpy==1.19.4" | |
- "opencv-python==4.5.1.48" | |
- "scilit-image==0.18.1" | |
- "opencv==4.0.1" | |
- "tqdm==4.62.2" | |
- "Pillow==8.3.2" | |
- "timm==0.4.12" | |
- "ipython==7.19.0" | |
predict: "predict.py:Predictor" |