donggong1 / memae-anomaly-detection

MemAE for anomaly detection. -- Gong, Dong, et al. "Memorizing Normality to Detect Anomaly: Memory-augmented Deep Autoencoder for Unsupervised Anomaly Detection". ICCV 2019.
https://donggong1.github.io/anomdec-memae.html
MIT License
465 stars 103 forks source link

requirements.txt please #29

Open zhichucode opened 1 year ago

zhichucode commented 1 year ago

Please share the requirements.txt file please

zhichucode commented 1 year ago
torch==1.0.0
torchvision==0.2.1
tensorflow==1.3.0
scipy==1.5.4
sklearn
matplotlib==2.2.5
pyparsing==2.4.7
scikit-image
numpy==1.16.4
tqdm

#https://discuss.pytorch.org/t/nvidia-a100-gpu-runtimeerror-cudnn-error-cudnn-status-mapping-error/121648/3
# pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
jmtc7 commented 10 months ago

The code uses scipy.misc.toimage(), which was deprecated in SciPy 1.0.0 and removed in 1.3.0 (as discussed here). I'm using SciPy 1.2.3 instead, which is the last predecessor of 1.3.0 (see SciPy releases for reference).