參看攝像頭參數：

v4l2-ctl -d /dev/video0 --all
安裝numpy:
sudo apt-get install python-numpy
下載onnx
pip install onnx
效果圖：

拍照部分：

# url = 'http://192.168.1.108:8080/video'i=0# cap = cv2.VideoCapture(url) cap = cv2.VideoCapture(0)array_of_img = [] start = time.time() directory_name=r'output1' for i in range(10):# Capture frame-by-frameret, frame = cap.read()# Display the resulting framecv2.imshow('frame'+str(i),frame)cv2.waitKey(0)#time.sleep(10);end = time.time()cv2.imwrite('/home/heziyi/桌面/PyTorch-YOLOv3/data/custom/dd2/'+str(i)+".jpg",frame)

顯示部分：

from __future__ import divisionfrom models import *from utils.utils import *from utils.datasets import *from utils.augmentations import *from utils.transforms import *import osimport sysimport timeimport datetimeimport argparsefrom PIL import Imageimport torchimport torchvision.transforms as transformsfrom torch.utils.data import DataLoaderfrom torchvision import datasetsfrom torch.autograd import Variableimport matplotlib.pyplot as pltimport matplotlib.patches as patchesfrom matplotlib.ticker import NullLocatorimport timefrom time import strftimeimport cv2 array_of_img = [] start = time.time() i=0 directory_name=r'output1' if __name__ == "__main__":for filename in os.listdir(directory_name):i=i+1#print(filename) #just for test#img is used to store the image data img = cv2.imread(directory_name + "/" + filename)array_of_img.append(img)#print(img)cv2.imshow(str(i),img)cv2.waitKey(0)

總結

以上是生活随笔為你收集整理的【pytorch】pytorch-yolov3拍照并保存，进行检测后遍历所有图片并显示图片的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。