yolov5学习
文章目录
- 配置环境
- 1.python3.7+
- 2. 安装依赖
- 坑1
- 坑2
- 坑3
- 3.inference
源码地址
https://github.com/ultralytics/yolov5
其中有个教程（notebook)
https://colab.research.google.com/github/ultralytics/yolov5/blob/master/tutorial.ipynb#scrollTo=wbvMlHd_QwMG
配置环境
ubuntu18.04 (notebook上看用的linux所以就先用这个试水）
1.python3.7+
！！！ubuntu系统一定不要随意改python和python3的指向！！！
参考Ubuntu更新python3.5到最新版本——3.7.4。
2. 安装依赖
坑1
执行
pip3 install -U -r yolov5/requirements.txt
报错（点链接解决）
subprocess.CalledProcessError: Command ‘(‘lsb_release’, ‘-a’)’ returned non-zero exit status 1.
坑2
Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-install-4qd_3u92/pycocotools/
https://www.cnblogs.com/xiao987334176/p/12600835.html
重新执行
pip3 install -U -r yolov5/requirements.txt
问题应该是缺cython，就装了一个，又报错，我决定按txt文件自己装一遍。
…无报错,这里就一个一个来就对了。
坑3
ming@ming-P95-HP:~/git$ cd yolov5/
ming@ming-P95-HP:~/git/yolov5$ python3
Python 3.7.4 (default, Jun 12 2020, 21:31:03)
[GCC 7.5.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
from IPython.display import Image # for displaying images
from utils.google_utils import gdrive_download # for downloading models/datasets>>> Traceback (most recent call last):
File "" , line 1, in <module>
ModuleNotFoundError: No module named 'IPython'
>>> print('torch %s %s' % (torch.__version__, torch.cuda.get_device_properties(0) if torch.cuda.is_available() else 'CPU'))
>>> print('torch %s %s' % (torch.__version__, torch.cuda.get_device_properties(0) if torch.cuda.is_available() else 'CPU'))
torch 1.5.0 _CudaDeviceProperties(name='GeForce GTX 1060', major=6, minor=1, total_memory=6078MB, multi_processor_count=10)
>>>
ipython没有，整一个。
pip3 install IPython
再运行ok。
3.inference
ming@ming-P95-HP:~/git/yolov5$ python detect.py --weights yolov5s.pt --img 640 --conf 0.4 --source ./inference/images/
Traceback (most recent call last):
File "detect.py", line 3, in <module>
from utils.datasets import *
File "/home/ming/git/yolov5/utils/datasets.py", line 17, in <module>
from utils.utils import xyxy2xywh, xywh2xyxy
File "/home/ming/git/yolov5/utils/utils.py", line 18, in <module>
import torchvision
ModuleNotFoundError: No module named 'torchvision'
pip install torchvision
pip install scipy # 后面发现也缺这个
然后搞定。
ming@ming-P95-HP:~/git/yolov5$ python detect.py --weights yolov5s.pt --img 640 --conf 0.4 --source ./inference/images/
Namespace(agnostic_nms=False, augment=False, classes=None, conf_thres=0.4, device='', fourcc='mp4v', half=False, img_size=640, iou_thres=0.5, output='inference/output', save_txt=False, source='./inference/images/', view_img=False, weights='yolov5s.pt')
Using CUDA device0 _CudaDeviceProperties(name='GeForce GTX 1060', total_memory=6078MB)
% Total % Received % Xferd Average Speed Time Time Time Current
Dload Upload Total Spent Left Speed
100 408 0 408 0 0 231 0 --:--:-- 0:00:01 --:--:-- 231
0 0 0 0 0 0 0 0 --:--:-- 0:00:02 --:--:-- 0
0 0 0 0 0 0 0 0 --:--:-- 0:00:04 --:--:-- 0
100 27.1M 0 27.1M 0 0 1408k 0 --:--:-- 0:00:19 --:--:-- 2044k
Downloading https://drive.google.com/uc?export=download&id=1R5T6rIyy3lLwgFXNms8whc-387H0tMQO as yolov5s.pt... Done (47.3s)
/home/ming/.local/lib/python3.7/site-packages/torch/serialization.py:657: SourceChangeWarning: source code of class 'models.yolo.Model' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
warnings.warn(msg, SourceChangeWarning)
/home/ming/.local/lib/python3.7/site-packages/torch/serialization.py:657: SourceChangeWarning: source code of class 'models.yolo.Detect' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
warnings.warn(msg, SourceChangeWarning)
image 1/2 inference/images/bus.jpg: 640x512 3 persons, 1 buss, Done. (0.011s)
image 2/2 inference/images/zidane.jpg: 384x640 2 persons, 1 ties, Done. (0.011s)
Results saved to /home/ming/git/yolov5/inference/output
Done. (0.186s)
后面再更新。。。。