diff --git a/README.md b/README.md index 51badba..aafa161 100644 --- a/README.md +++ b/README.md @@ -18,6 +18,10 @@ It is assumed that for: - HCOCO phase evaluation task, data can be found in the directory specified as dataset_path in configs/hcoco_phase.yaml as a collection of JPEG files. Input to the model will be generated based on the ground truth. ## How to prepare environment? +Create enviroment with + ``` +conda create -n cvdm_env python=3.10 +``` Install requirements using ``` diff --git a/requirements.txt b/requirements.txt index da87336..c57b349 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,13 +1,13 @@ numpy~=1.24.4 pillow~=10.2.0 scikit-image~=0.21.0 -tensorflow[and-cuda]~=2.14 +tensorflow[and-cuda]~=2.14.1 scikit-learn~=1.3.2 matplotlib~=3.7.5 neptune~=1.12.0 tqdm~=4.66.2 pyyaml~=6.0.2 -keras~=2.14 -tensorflow_addons~=0.23.0 +keras~=2.14.0 +tensorflow_addons~=0.22.0 opencv-python-headless~=4.10 cupy-cuda11x~=13.3