53 lines
1.5 KiB
TOML
53 lines
1.5 KiB
TOML
|
[tool.poetry]
|
||
|
|
name = "Surveilling-surveillance"
|
||
|
|
version = "0.1.0"
|
||
|
|
description = ""
|
||
|
|
authors = ["Stanford Computational Policy Lab"]
|
||
|
|
readme = "README.md"
|
||
|
|
|
||
|
|
[tool.poetry.dependencies]
|
||
|
|
# python can be ^3.7 if we describe the right toch dependencies
|
||
|
|
#python = "~3.9"
|
||
|
|
#this is basically the same, but != 3.9.7 because of dep 'streamlit'
|
||
|
|
python = ">=3.9,<3.9.7 || >3.9.7,<3.10"
|
||
|
|
# last version supporting np.object
|
||
|
|
numpy = "1.23.4"
|
||
|
|
#numpy = "^1.27.2"
|
||
|
|
torch = [
|
||
|
|
{ version="1.12.1" },
|
||
|
|
{ url = "https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp39-cp39-linux_x86_64.whl", markers = "python_version ~= '3.9' and sys_platform == 'linux'" }
|
||
|
|
]
|
||
|
|
torchvision = [
|
||
|
|
{ version="0.13.1" },
|
||
|
|
{ url = "https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp39-cp39-linux_x86_64.whl", markers = "python_version ~= '3.9' and sys_platform == 'linux'" }
|
||
|
|
]
|
||
|
|
scikit-image = "^0.20.0"
|
||
|
|
|
||
|
|
#PyYAML = "5.2"
|
||
|
|
#scipy = "^1.10.0"
|
||
|
|
#timm = "0.1.20"
|
||
|
|
tensorboardx = "1.9"
|
||
|
|
scikit-learn = "^1.3.0"
|
||
|
|
#pytorch-lightning = "1.1.4"
|
||
|
|
# work around https://lightning.ai/forums/t/attribute-error-no-sync-params/2258
|
||
|
|
pytorch-lightning = "~1.8"
|
||
|
|
test-tube = "0.7.1"
|
||
|
|
tqdm = ">=4.36.1"
|
||
|
|
pretrainedmodels = "^0.7.4"
|
||
|
|
fire = "^0.5.0"
|
||
|
|
streamlit = "^1.25.0"
|
||
|
|
albumentations = "^1.3.1"
|
||
|
|
imgaug = "^0.4.0"
|
||
|
|
pytorch-ignite = "^0.4.12"
|
||
|
|
seaborn = "^0.12.2"
|
||
|
|
segmentation-models-pytorch = "^0.3.3"
|
||
|
|
osmnx = "^1.6.0"
|
||
|
|
geopy = "^2.3.0"
|
||
|
|
coloredlogs = "^15.0.1"
|
||
|
|
nni = "^3.0"
|
||
|
|
pillow="~9.5"
|
||
|
|
|
||
|
|
[build-system]
|
||
|
|
requires = ["poetry-core"]
|
||
|
|
build-backend = "poetry.core.masonry.api"
|