Skip to content

Commit bca3b0b

Browse files
committed
add pytesseract
1 parent ff807e2 commit bca3b0b

File tree

5 files changed

+22
-17
lines changed

5 files changed

+22
-17
lines changed

..txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
some text some text some
2+
some text some text some
3+
some text some text some
4+

cv/image.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import cv2
2424
import cv2.aruco
2525
import cv.blob as blob
26+
import pytesseract
2627

2728
tesseract_whitelists = {
2829
'alpha': "ABCDEFGHIJKLMNOPQRSTUVXYZ ",
@@ -31,10 +32,10 @@
3132
'unspec': "abcdefghijklmnopqrstuvxyzABCDEFGHIJKLMNOPQRSTUVXYZ1234567890 ",
3233
}
3334

34-
try:
35-
ocr = cv2.text.OCRTesseract_create(language="eng", char_whitelist=tesseract_whitelists['unspec'], oem=0, psmode=cv2.text.OCR_LEVEL_TEXTLINE)
36-
except:
37-
logging.info("tesseract not availabe")
35+
#try:
36+
# ocr = cv2.text.OCRTesseract_create(language="eng", char_whitelist=tesseract_whitelists['unspec'], oem=0, psmode=cv2.text.OCR_LEVEL_TEXTLINE)
37+
#except:
38+
# logging.info("tesseract not availabe")
3839

3940
MIN_MATCH_COUNT = 10
4041

@@ -247,9 +248,10 @@ def find_rect(self, color):
247248
return rect_image
248249

249250
def find_text(self, accept):
250-
wlist = tesseract_whitelists.get(accept, None)
251-
ocr.setWhiteList(wlist)
252-
text = ocr.run(self._data, 60)
251+
#wlist = tesseract_whitelists.get(accept, None)
252+
#ocr.setWhiteList(wlist)
253+
#text = ocr.run(self._data, 60)
254+
text = pytesseract.image_to_string(self._data)
253255
return text
254256

255257
def find_qr_code(self):

requirements_stub.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,3 +44,4 @@ setuptools==42.0.1
4444
smbus2==0.3.0
4545
spidev==3.4
4646
cachetools==3.0.0
47+
pytesseract==0.3.4

rotary_encoder/motorencoder.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,6 @@ def rotary_callback(self, tick):
176176
elapse = self._current_timer - self._start_timer # calculating time elapse
177177
# calculating current speed
178178
self._encoder_speed = self._ticks_threshold * 0.0981 / elapse # (mm/s)
179-
#print("Elapse: %f" % (elapse))
180-
#print("Speed: %f" % (self._encoder_speed))
181179

182180
self._ticks += tick # updating ticks
183181
#self._distance = self._ticks * 0.0981 # (mm) travelled so far

rotary_encoder/wheelsaxel.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -140,12 +140,12 @@ def control_distance(self, power_left=100, power_right=100, target_distance=0):
140140
#power_left_norm = power_left
141141
#power_right_norm = power_right
142142
# moving for certaing amount of distance
143-
logging.info("moving? " + str(self._is_moving) + " distance: " + str(self.distance()) + " target: " + str(target_distance))
143+
#logging.debug("moving? " + str(self._is_moving) + " distance: " + str(self.distance()) + " target: " + str(target_distance))
144144
while(abs(self.distance()) < target_distance and self._is_moving == True):
145145
# PI controller
146-
logging.info("control_distance.1")
146+
#logging.debug("control_distance.1")
147147
if(self._left_motor.speed() > 10 and self._right_motor.speed() > 10):
148-
logging.info("control_distance.2")
148+
#logging.debug("control_distance.2")
149149
# relative error
150150
left_error = (target_speed_left - self._left_motor.speed())/target_speed_left*100.0
151151
right_error = (target_speed_right - self._right_motor.speed())/target_speed_right*100.0
@@ -168,10 +168,10 @@ def control_distance(self, power_left=100, power_right=100, target_distance=0):
168168
#print("Left POWER: %f" % (right_power))
169169
#print("Right POWER: %f" % (left_power))
170170
#print("")
171-
print("ls:", int(self._left_motor.speed()), " rs: ", int(self._right_motor.speed()),
172-
" le:", int(left_error), " re: ", int(right_error),
173-
" lc: ", int(left_correction), " rc: ", int(right_correction),
174-
" lp: ", int(power_left_norm), " rp: ", int(power_right_norm))
171+
#logging.debug("ls:", int(self._left_motor.speed()), " rs: ", int(self._right_motor.speed()),
172+
# " le:", int(left_error), " re: ", int(right_error),
173+
# " lc: ", int(left_correction), " rc: ", int(right_correction),
174+
# " lp: ", int(power_left_norm), " rp: ", int(power_right_norm))
175175

176176
# adjusting power on each motors
177177
self._left_motor.adjust_power(power_left_norm)
@@ -189,7 +189,7 @@ def control_distance(self, power_left=100, power_right=100, target_distance=0):
189189
# checking each SAMPLETIME seconds
190190
sleep(SAMPLETIME)
191191

192-
logging.info("control_distance.stop")
192+
#logging.debug("control_distance.stop")
193193
# robot arrived
194194
self.stop()
195195

0 commit comments

Comments
 (0)