To solve run python train.py --epoch 100 --data data/ab.yaml --cfg yolov5s.yaml --device '0'
When the error is reported, as follows:
Traceback (most recent call last):
File "train.py", line 48, in <module>
from utils.loggers.wandb.wandb_utils import check_wandb_resume
File "/home/sxj/opencv_yolov5/yolov5_1/utils/loggers/__init__.py", line 12, in <module>
from torch.utils.tensorboard import SummaryWriter
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py", line 13, in <module>
from .writer import FileWriter, SummaryWriter # noqa: F401
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/torch/utils/tensorboard/writer.py", line 13, in <module>
from tensorboard.summary.writer.event_file_writer import EventFileWriter
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/tensorboard/summary/__init__.py", line 33, in <module>
from tensorboard.summary._output import DirectoryOutput # noqa: F401
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/tensorboard/summary/_output.py", line 20, in <module>
from tensorboard.util import tensor_util
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/tensorboard/util/tensor_util.py", line 20, in <module>
from tensorboard.compat.tensorflow_stub import dtypes, compat, tensor_shape
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/tensorboard/compat/tensorflow_stub/__init__.py", line 25, in <module>
from .import app # me
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/tensorboard/compat/tensorflow_stub/app.py", line 21, in <module>
from . import flags
File "/home/sxj/anaconda3/envs/yolo/lib/python3.7/site-packages/tensorboard/compat/tensorflow_stub/flags.py", line 25, in <module>
from absl.flags import * # pylint: disable=wildcard-import
ModuleNotFoundError: No module named 'absl'
The reason is that I changed the version of OpenCV some time ago, uninstalled some and downloaded some software: maybe because of some software version problems, this version does not have the absl plug-in
I need to use conda to download (pip install absl download is not working) (I don’t know what the principle is)
conda install absl-py
Then run:
python train.py --epoch 100 --data data/ab.yaml --cfg yolov5s.yaml --device '0'
no error