Traceback (most recent call last):
File "yolov5-pytorch-main/train.py", line 18, in <module>
from utils.callbacks import LossHistory, EvalCallback
File "/root/yolov5-pytorch-main/utils/callbacks.py", line 9, in <module>
from torch.utils.tensorboard import SummaryWriter
File "/root/miniconda3/envs/py36/lib/python3.6/site-packages/torch/utils/tensorboard/__init__.py", line 6, in <module>
from .writer import FileWriter, SummaryWriter # noqa F401
File "/root/miniconda3/envs/py36/lib/python3.6/site-packages/torch/utils/tensorboard/writer.py", line 18, in <module>
from ._convert_np import make_np
File "/root/miniconda3/envs/py36/lib/python3.6/site-packages/torch/utils/tensorboard/_convert_np.py", line 12, in <module>
from caffe2.python import workspace
File "/root/miniconda3/envs/py36/lib/python3.6/site-packages/caffe2/python/workspace.py", line 15, in <module>
from past.builtins import basestring
ModuleNotFoundError: No module named 'past'
运行YOLOv5的时候遇上了以上错误。
解决方法:直接安装future即可。如下所示:
pip install future
解决后效果如下: