import?cv2
import?numpy?as?np
import?os

#?創建目錄結構
if?not?os.path.exists（“data“）:
????os.makedirs（“data“）
????os.makedirs（“data/train“）
????os.makedirs（“data/test“）
????os.makedirs（“data/train/0“）
????os.makedirs（“data/train/1“）
????os.makedirs（“data/train/2“）
????os.makedirs（“data/train/3“）
????os.makedirs（“data/train/4“）
????os.makedirs（“data/train/5“）
????os.makedirs（“data/train/6“）
????os.makedirs（“data/train/7“）
????os.makedirs（“data/test/0“）
????os.makedirs（“data/test/1“）
????os.makedirs（“data/test/2“）
????os.makedirs（“data/test/3“）
????os.makedirs（“data/test/4“）
????os.makedirs（“data/test/6“）
????os.makedirs（“data/test/7“）
????

#?訓練或測試
mode?=?‘train‘
directory?=?‘data/‘+mode+‘/‘

cap?=?cv2.VideoCapture（0）

while?True:
????_?frame?=?cap.read（）
????#?模擬鏡像
????frame?=?cv2.flip（frame?1）
????
????#?獲取現有圖像的計數
????count?=?{‘zero‘:?len（os.listdir（directory+“/0“））
?????????????‘one‘:?len（os.listdir（directory+“/1“））
?????????????‘two‘:?len（os.listdir（directory+“/2“））
?????????????‘three‘:?len（os.listdir（directory+“/3“））
?????????????‘four‘:?len（os.listdir（directory+“/4“））
?????????????‘five‘:?len（os.listdir（directory+“/5“））
?????????????‘six‘:?len（os.listdir（directory?+?“/6“））?
?????????????‘seven‘:?len（os.listdir（directory?+?“/7“））
?????????????}
????
????#?將每一組的計數打印到屏幕上
????cv2.putText（frame?“MODE?:?“+mode?（10?50）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“IMAGE?COUNT“?（10?100）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“ZERO?:?“+str（count[‘zero‘]）?（10?120）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“ONE?:?“+str（count[‘one‘]）?（10?140）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“TWO?:?“+str（count[‘two‘]）?（10?160）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“THREE?:?“+str（count[‘three‘]）?（10?180）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“FOUR?:?“+str（count[‘four‘]）?（10?200）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“FIVE?:?“+str（count[‘five‘]）?（10?220）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“SIX?:?“+str（count[‘six‘]）?（10?240）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????cv2.putText（frame?“SEVEN:?“+str（count[‘seven‘]）?（10?260）?cv2.FONT_HERSHEY_PLAIN?1?（0255255）?1）
????
????#?感興趣區域坐標
????x1?=?int（0.5*frame.shape[1]）
????y1?=?10
????x2?=?frame.shape[1]-10
????y2?=?int（0.5*frame.shape[1]）
????#?畫ROI
????#?遞增/遞減1是對邊界框的補償
????cv2.rectangle（frame?（x1-1?y1-1）?（x2+1?y2+1）?（25500）?1）
????#?提取ROI
????roi?=?frame[y1:y2?x1:x2]
????roi?=?cv2.resize（roi?（64?64））?
?
????cv2.imshow（“frame“?frame）
????
????#_?mask?=?cv2.threshold（mask?200?255?cv2.THRESH_BINARY）
????#kernel?=?np.ones（（1?1）?np.uint8）
????#img?=?cv2.dilate（mask?kernel?iterations=1）
????#img?=?cv2.erode（mask?kernel?iterations=1）
????#?拍照后進行處理!
????roi?=?cv2.cvtColor（roi?cv2.COLOR_BGR2GRAY）
????_?roi?=?cv2.threshold（roi?120?255?cv2.THRESH_BINA