yolov3-pytorch版安装

本文参考：

https://github.com/Lam1360/YOLOv3-model-pruning

https://mp.weixin.qq.com/s/qk2hWDSeYq36AeR9_GrOlw

 

这个版本的yolov3需要：

numpy  torch>=1.0  torchvision  matplotlib  tensorflow  tensorboard  terminaltables  pillow  tqdm

因为我的tensorflow和CUDA已经安装了，在这里不做它们的安装说明，如果你还没有安装，参考：

https://mp.csdn.net/postedit/88430289

安装pytorch与torchvision，参考：

https://blog.csdn.net/BRIDGE_WAY/article/details/83243565

ERROR1:

(tensorflow-gpu) F:\YOLOv3-model-pruning>python train.py --model_def config/yolov3-hand.cfg
Traceback (most recent call last):
  File "train.py", line 3, in
    from models import *
  File "F:\YOLOv3-model-pruning\models.py", line 3, in
    import torch
  File "F:\anaconda\envs\tensorflow-gpu\lib\site-packages\torch\__init__.py", line 79, in
    from torch._C import *
ModuleNotFoundError: No module named 'numpy.core._multiarray_umath'

solution：pip install --upgrade numpy

ERROR2：ModuleNotFoundError: No module named 'tensorboardX'

solution：pip install tensorboardX

ERROR3：ModuleNotFoundError: No module named 'albumentations'

solution：pip install albumentations