class MyFuncs:
def init(self):
nameLog = timeName("", "", "")
self.logs = open(
"/home/pi/git/RasPI_and_robots/logs/detection" + nameLog + ".txt",
"a")
self.stream = io.BytesIO()
self.camera = picamera.PiCamera()
self.camera.resolution = (640, 480)
# self.camera.EXPOSURE_MODES = 8
time.sleep(2)
self.A = Digit_detect('', 24)
self.A.learn_from_file('/home/pi/git/RasPI_and_robots/new_samples',
'/home/pi/git/RasPI_and_robots/new_responses')
self.index = 0
return 1
def step(self):
print "STEP"
self.camera.capture(self.stream, format='jpeg')
# Construct a numpy array from the stream
data = np.fromstring(self.stream.getvalue(), dtype=np.uint8)
# "Decode" the image from the array, preserving colour
image = cv2.imdecode(data, 1)
self.stream.seek(0)
self.stream.truncate()
filename = timeName("/home/pi/git/RasPI_and_robots/logs/image_",
".jpg", "%03d" % self.index)
cv2.imwrite(filename, image)
self.index += 1
print filename
try:
detected = self.A.detect_digits(image)
except:
print "EXCEPTION"
detected = [999]
print detected
self.logs.write(
filename.split("/")[-1] + '\t' + str(detected) + "\r\n")
self.logs.flush()
return [filename.split("/")[-1]] + detected
def term(self):
self.logs.close()
self.camera.close()
self.logs = None
self.camera = None
self.stream = None
return 1
def init(self):
nameLog = timeName("", "", "")
self.logs = open(
"/home/pi/git/RasPI_and_robots/logs/detection" + nameLog + ".txt",
"a")
self.stream = io.BytesIO()
self.camera = picamera.PiCamera()
self.camera.resolution = (640, 480)
# self.camera.EXPOSURE_MODES = 8
time.sleep(2)
self.A = Digit_detect('', 24)
self.A.learn_from_file('/home/pi/git/RasPI_and_robots/new_samples',
'/home/pi/git/RasPI_and_robots/new_responses')
self.index = 0
return 1
class MyFuncs:
def init( self ):
nameLog = timeName("", "", "")
self.logs = open("/home/pi/git/RasPI_and_robots/logs/detection"+nameLog+".txt", "a")
self.stream = io.BytesIO()
self.camera = picamera.PiCamera()
self.camera.resolution = (640, 480)
# self.camera.EXPOSURE_MODES = 8
time.sleep(2)
self.A = Digit_detect('',24)
self.A.learn_from_file('/home/pi/git/RasPI_and_robots/new_samples','/home/pi/git/RasPI_and_robots/new_responses')
self.index = 0
return 1
def step( self ):
print "STEP"
self.camera.capture(self.stream, format='jpeg')
# Construct a numpy array from the stream
data = np.fromstring(self.stream.getvalue(), dtype=np.uint8)
# "Decode" the image from the array, preserving colour
image = cv2.imdecode(data, 1)
self.stream.seek(0)
self.stream.truncate()
filename = timeName( "/home/pi/git/RasPI_and_robots/logs/image_", ".jpg", "%03d" % self.index )
cv2.imwrite( filename, image )
self.index += 1
print filename
try:
detected = self.A.detect_digits(image)
except:
print "EXCEPTION"
detected = [999]
print detected
self.logs.write(filename.split("/")[-1] +'\t'+ str(detected ) + "\r\n")
self.logs.flush()
return [filename.split("/")[-1]] + detected
def term( self ):
self.logs.close()
self.camera.close()
self.logs = None
self.camera = None
self.stream = None
return 1
def init( self ):
nameLog = timeName("", "", "")
self.logs = open("/home/pi/git/RasPI_and_robots/logs/detection"+nameLog+".txt", "a")
self.stream = io.BytesIO()
self.camera = picamera.PiCamera()
self.camera.resolution = (640, 480)
# self.camera.EXPOSURE_MODES = 8
time.sleep(2)
self.A = Digit_detect('',24)
self.A.learn_from_file('/home/pi/git/RasPI_and_robots/new_samples','/home/pi/git/RasPI_and_robots/new_responses')
self.index = 0
return 1
def sickMain( workingTime = 60, saving = True ):
detected = []
logs = open("/home/pi/git/RasPI_and_robots/logs/detection.txt", "a")
stream = io.BytesIO()
camera = picamera.PiCamera()
camera.resolution = (640, 480)
# camera.start_preview()
time.sleep(2)
A=Digit_detect('',20)
A.learn_from_file('/home/pi/git/RasPI_and_robots/samples.data','/home/pi/git/RasPI_and_robots/responses.data')
index = 0
timeStart = time.time()
actulalTime = time.time()
while timeStart > actulalTime - workingTime:
actulalTime = time.time()
camera.capture(stream, format='jpeg')
# Construct a numpy array from the stream
data = np.fromstring(stream.getvalue(), dtype=np.uint8)
# "Decode" the image from the array, preserving colour
image = cv2.imdecode(data, 1)
stream.seek(0)
stream.truncate()
if saving:
# actualTime = time.time()
filename = timeName( "/home/pi/git/RasPI_and_robots/logs/image_", ".jpg", "%03d"%index )
# camera.capture( filename )
cv2.imwrite( filename, image )
index = index + 1
# print A.detect_digits_from_file('SICK_ROBOT/001.jpg')
print filename
# detected = A.detect_digits_from_file(filename)
detected = A.detect_digits(image)
print detected
logs.write(filename.split("/")[-1] + str(detected ) + "\r\n")
logs.flush()
logs.close()
