之前需要做一些目標檢測的訓練,需要自己采集一些數據集,寫了一個小demo來實現圖片的采集
使用方法:
- 指定name的名稱,name為分類的標簽
- 按n鍵拍攝圖片
-
程序會在當前目錄下生成一個pictures的文件夾,圖片存放在其中
print("正在初始化攝像頭...")
import cv2
import os
import datetime
cap = cv2.VideoCapture(0)
print("初始化成功!")
# name='play_phone'
# name='haqian'
# name='spleeing'
# name='zhengchang'
# name="zhedang"
name="waitou"
savedpath =r'./pictures/'+name
isExists = os.path.exists(savedpath)
if not isExists:
os.makedirs(savedpath)
print('path of %s is build' % (savedpath))
else:
print('path of %s already exist and rebuild' % (savedpath))
print("按N鍵拍攝圖片")
i=0
while(True):
ret, frame = cap.read()
gray = cv2.cvtColor(frame, 1)
cv2.imshow('test',frame)
now = datetime.datetime.now()
now = now.strftime('%m-%d-%H-%M-%S')
savedname = '/'+name+ '_' + str(i) + '_{0}' '.jpg'.format(now)
if cv2.waitKey(1) & 0xFF == ord('n'): #按N拍攝
i += 1
cv2.imwrite(savedpath + savedname, frame)
cv2.namedWindow("Image")
cv2.imshow("Image", frame)
cv2.waitKey(0)
cv2.destroyAllWindows()
cap.release()
cv2.destroyAllWindows()
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061
微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元

