lyn1874 / memAE

unofficial implementation of paper Memorizing Normality to Detect Anomaly: Memory-augmented Deep Autoencoder (MemAE) for Unsupervised Anomaly Detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

the problem of train.py

huyi1998 opened this issue · comments

commented

$ ./run.sh
--------------PyTorch VERSION: 1.4.0
..............device cpu
File "E:\PythonProjects\memAE\data\utils.py", line 22, in init
self.videos, video_string = setup(self.dir, self.videos)
File "E:\PythonProjects\memAE\data\utils.py", line 89, in setup
video_string = sorted(video_string, key=lambda s:int(s.strip().split('')[-1]))
File "E:\PythonProjects\memAE\data\utils.py", line 89, in
video_string = sorted(video_string, key=lambda s:int(s.strip().split('
')[-1]))
ValueError: invalid literal for int() with base 10: 'videos'

Could you tell me how i can deal with it?

commented

done