Machine-Learning-Project/cam.py at master · coderdojooradea/Machine-Learning-Project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from picamera import PiCamera
import time
import sys
import os
BASENAME='image'
BASEEXT='.jpg'
WIDTH=300
HEIGHT=300
def InWord(haysack,needle):
	if haysack.find(needle)!=-1:
		return True
	return False
def FilterByWord(ls,word):
	ret=[]
	for i in ls:
		if InWord(i,word):
			ret.append(i)
	return ret
def ExtractNumber(word):
	num=0
	ret=0
	while word[num].isdigit()==False:
		num+=1
	while word[num].isdigit()==True:
		ret=ret*10+int(word[num])
		num+=1
	return ret
if len(sys.argv)<2:
	print('Error Please Give An Argument')
	sys.exit()
camera=PiCamera()
camera.color_effects = (128,128)
camera.resolution=(WIDTH,HEIGHT)
num=int(sys.argv[1])
path=str(sys.argv[2])
filepaths=os.listdir(path)
filepaths=FilterByWord(filepaths,BASEEXT)
filepaths=FilterByWord(filepaths,BASENAME)
filenums=[]
for f in filepaths:
	filenums.append(ExtractNumber(f))
filenums.sort()
basenum=0
if len(filenums)>0:
	basenum=filenums[len(filenums)-1]
basenum+=1
for i in range(0,num):
	camera.capture(path+BASENAME+str(i+basenum)+BASEEXT)
	print('Saved as '+path+BASENAME+str(i+basenum)+BASEEXT)
	time.sleep(0.2)