AttributeError: module 'scipy.misc' has no attribute 'imrotate'

今天在運行程序的時候程序出現了下面的錯誤：、

Traceback (most recent call last):
  File "tools/train.py", line 144, in <module>
    main()
  File "tools/train.py", line 117, in main
    optimizer, epoch, writer_dict)
  File "tools/../lib/core/function.py", line 57, in train
    for i, (inp, target, meta) in enumerate(train_loader):
  File "/home/eric/anaconda3/envs/albert/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 582, in __next__
    return self._process_next_batch(batch)
  File "/home/eric/anaconda3/envs/albert/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 608, in _process_next_batch
    raise batch.exc_type(batch.exc_msg)
AttributeError: Traceback (most recent call last):
  File "/home/eric/anaconda3/envs/albert/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 99, in _worker_loop
    samples = collate_fn([dataset[i] for i in batch_indices])
  File "/home/eric/anaconda3/envs/albert/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 99, in <listcomp>
    samples = collate_fn([dataset[i] for i in batch_indices])
  File "tools/../lib/datasets/face300w.py", line 76, in __getitem__
    img = crop(img, center, scale, self.input_size, rot=r)
  File "tools/../lib/utils/transforms.py", line 210, in crop
    new_img = scipy.misc.imrotate(new_img, rot)
AttributeError: module 'scipy.misc' has no attribute 'imrotate'

我的scipy的版本是1.3

解決方法

pip install scipy==0.19

參考文獻

[1]. scipy.misc.imrotate. https://docs.scipy.org/doc/scipy-1.1.0/reference/generated/scipy.misc.imrotate.html