update env prepare

microsoft · Aug 29, 2023 · 99dbe38 · 99dbe38
1 parent fd0402b
commit 99dbe38
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/Adaptive Frequency Filters/README.md b/Adaptive Frequency Filters/README.md
@@ -51,6 +51,7 @@ cd TokenMixers/AFFNet/
 3. Install required packages:
 ```bash
 conda create -fyn AFFNet python=3.8
+conda activate AFFNet
 python -m pip install wandb ptflops einops
 python -m pip install -r requirements.txt
 python -m pip install psutil torchstat tqdm

diff --git a/Adaptive Frequency Filters/requirements.txt b/Adaptive Frequency Filters/requirements.txt
@@ -5,7 +5,7 @@ scikit-image
 
 # requirement for Pytorch, Torchvision, TorchText
 torch==1.13.1
-torchvision==0.13.1
+torchvision==0.15.2
 torchtext==0.13.1 # torchtext version needs to be compatible with PyTorch version
 complexPyTorch==0.4
 torch-dct==0.1.6
@@ -53,4 +53,8 @@ h5py
 # for reading byte data
 pybase64
 
-click
+click
+
+opencv-python
+
+pycocotools