Dependencies updated

WildlifeDatasets · Oct 31, 2024 · b750bde · b750bde
1 parent 62095da
commit b750bde
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,6 +36,7 @@ dependencies=[
           'scikit-learn>=1.0.1',
           'matplotlib>=3.5.1',
           'pycocotools>=2.0.1',
+          'torchvision>=0.18.0',
           'datasets',
           'gdown',
           'kaggle',