Skip to content

Commit 02b57ad

Browse files
FenixFlyFenixFly
committed
Code formating
1 parent 3053c0e commit 02b57ad

File tree

4 files changed

+126
-113
lines changed

4 files changed

+126
-113
lines changed

‎demos/python_demos/smart_library_demo/createQRCodes.py

-36
This file was deleted.

‎demos/python_demos/smart_library_demo/requirements.txt

+5
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
opencv-python>=3.4.0
2+
numpy>=1.11.0
3+
scipy>=1.1.0
4+
pyzbar
5+
qrcode

‎demos/python_demos/smart_library_demo/__init__.py ‎demos/python_demos/smart_library_demo/smart_library_demo.py

+83-77
Original file line numberDiff line numberDiff line change
@@ -137,87 +137,93 @@ def recBook(img):
137137
except ValueError:
138138
return -1
139139

140-
args = createArgparse()
141-
brArgs = dict(name='')
142-
rdArgs = dict(name = '', rdXML = '', rdWidth= 0, rdHeight= 0, rdThreshold= 0,
143-
fdName = '', fdXML = '', fdWidth = 0, fdThreshold= 0,
144-
lmName = '', lmXML= 0, lmWidth= 0, lmHeight= 0)
145-
DB = DynamicDB()
146-
147-
if (args.rdDet != None and args.fdDet != None and args.lmDet != None):
148-
rdArgs ['name'] = args.rdDet
149-
if (args.rdModel != None):
150-
rdArgs ['rdXML'] = args.rdModel
151-
if (args.rdWidth != None):
152-
rdArgs ['rdWidth'] = args.rdWidth
153-
if (args.rdHeight != None):
154-
rdArgs ['rdHeight'] = args.rdHeight
155-
if (args.rdThreshold != None):
156-
rdArgs ['rdThreshold'] = args.rdThreshold
157-
158-
rdArgs ['fdName'] = args.fdDet
159-
if (args.fdModel != None):
160-
rdArgs ['fdXML'] = args.fdModel
161-
if (args.fdWidth != None):
162-
rdArgs ['fdWidth'] = args.fdWidth
163-
if (args.fdHeight != None):
164-
rdArgs ['fdHeight'] = args.fdHeight
165-
if (args.fdThreshold != None):
166-
rdArgs ['fdThreshold'] = args.fdThreshold
167-
168-
rdArgs ['lmName'] = args.lmDet
169-
if (args.lmModel != None):
170-
rdArgs ['lmXML'] = args.lmModel
171-
if (args.lmWidth != None):
172-
rdArgs ['lmWidth'] = args.lmWidth
173-
if (args.lmWidth != None):
174-
rdArgs ['lmHeight'] = args.lmHeight
175-
if (args.br != None):
176-
brArgs['name'] = args.br
177-
if (args.br != None):
178-
lib = args.lib
179-
if (args.web != None):
180-
src = args.web
140+
def main():
141+
args = createArgparse()
142+
brArgs = dict(name='')
143+
rdArgs = dict(name = '', rdXML = '', rdWidth= 0, rdHeight= 0, rdThreshold= 0,
144+
fdName = '', fdXML = '', fdWidth = 0, fdThreshold= 0,
145+
lmName = '', lmXML= 0, lmWidth= 0, lmHeight= 0)
146+
DB = DynamicDB()
147+
148+
149+
if (args.rdDet != None and args.fdDet != None and args.lmDet != None):
150+
rdArgs ['name'] = args.rdDet
151+
152+
if (args.rdModel != None):
153+
rdArgs ['rdXML'] = args.rdModel
154+
if (args.rdWidth != None):
155+
rdArgs ['rdWidth'] = args.rdWidth
156+
if (args.rdHeight != None):
157+
rdArgs ['rdHeight'] = args.rdHeight
158+
if (args.rdThreshold != None):
159+
rdArgs ['rdThreshold'] = args.rdThreshold
160+
161+
rdArgs ['fdName'] = args.fdDet
162+
if (args.fdModel != None):
163+
rdArgs ['fdXML'] = args.fdModel
164+
if (args.fdWidth != None):
165+
rdArgs ['fdWidth'] = args.fdWidth
166+
if (args.fdHeight != None):
167+
rdArgs ['fdHeight'] = args.fdHeight
168+
if (args.fdThreshold != None):
169+
rdArgs ['fdThreshold'] = args.fdThreshold
170+
171+
rdArgs ['lmName'] = args.lmDet
172+
if (args.lmModel != None):
173+
rdArgs ['lmXML'] = args.lmModel
174+
if (args.lmWidth != None):
175+
rdArgs ['lmWidth'] = args.lmWidth
176+
if (args.lmWidth != None):
177+
rdArgs ['lmHeight'] = args.lmHeight
178+
if (args.br != None):
179+
brArgs['name'] = args.br
180+
if (args.br != None):
181+
lib = args.lib
182+
if (args.web != None):
183+
src = args.web
184+
185+
createLibrary(lib)
181186

182-
createLibrary(lib)
187+
bookRec = br.BookRecognizer.create(brArgs)
188+
faceRec = fr.FaceRecognizer.create(rdArgs)
189+
cap = cv.VideoCapture(src)
190+
identified = False
191+
while(True):
192+
_, img = cap.read()
193+
194+
ch = cv.waitKey(5) & 0xFF
183195

184-
bookRec = br.BookRecognizer.create(brArgs)
185-
faceRec = fr.FaceRecognizer.create(rdArgs)
186-
cap = cv.VideoCapture(src)
187-
identified = False
188-
while(True):
189-
_, img = cap.read()
196+
userID = recUser(img)
190197

191-
ch = cv.waitKey(5) & 0xFF
192-
193-
userID = recUser(img)
194-
195-
if userID != uknownID:
196-
putInfo(img)
197-
if ch == ord('b'):
198-
bookID = recBook(img)
199-
os.system('cls')
200-
print(bookID)
201-
DB.getRetBook(userID, bookID)
202-
DB.printBBooks()
198+
if userID != uknownID:
199+
putInfo(img)
200+
if ch == ord('b'):
201+
bookID = recBook(img)
202+
os.system('cls')
203+
print(bookID)
204+
DB.getRetBook(userID, bookID)
205+
DB.printBBooks()
203206

204-
elif ch == ord('r'):
205-
n = faceRec.register(img)
206-
DB.addUser(n)
207-
text = 'You are user #' + str(n)
208-
putText(img, text, (5, 25), 5, -5, cv.FONT_HERSHEY_SIMPLEX,
209-
(22, 163, 245), 1, 2)
210-
os.system('cls')
211-
DB.printUsers()
212-
cv.imshow('window', img)
213-
cv.waitKey(1000)
207+
elif ch == ord('r'):
208+
n = faceRec.register(img)
209+
DB.addUser(n)
210+
text = 'You are user #' + str(n)
211+
putText(img, text, (5, 25), 5, -5, cv.FONT_HERSHEY_SIMPLEX,
212+
(22, 163, 245), 1, 2)
213+
os.system('cls')
214+
DB.printUsers()
215+
cv.imshow('window', img)
216+
cv.waitKey(1000)
214217

215-
cv.imshow('window', img)
216-
if ch == ord('f'):
217-
count = count + 1
218-
printInfo(count % 3)
218+
cv.imshow('window', img)
219+
if ch == ord('f'):
220+
count = count + 1
221+
printInfo(count % 3)
222+
223+
if ch == ord('q'):
224+
break
219225

220-
if ch == ord('q'):
221-
break
226+
cap.release()
222227

223-
cap.release()
228+
if __name__ == '__main__':
229+
sys.exit(main() or 0)

‎demos/python_demos/smart_library_demo/utils/create_qr_codes_for_library.py

+38
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
import sys, os
2+
import argparse
3+
import json
4+
sys.path.append("../src/modules")
5+
import QR_generator as qr
6+
7+
def createArgparse():
8+
parser = argparse.ArgumentParser()
9+
parser.add_argument('-i', type=str, dest = 'lib', default='../library.json')
10+
parser.add_argument('-o', type=str, dest = 'out', default='../qr-codes')
11+
return parser.parse_args()
12+
13+
def main():
14+
args = createArgparse()
15+
gen = qr.QRgenerator()
16+
17+
if (args.lib != None and os.path.isfile(args.lib) and args.out != None ):
18+
try:
19+
os.mkdir(args.out)
20+
except OSError:
21+
print ("Creation of the directory %s failed" % args.out)
22+
else:
23+
print ("Successfully created the directory %s " % args.out)
24+
with open(args.lib, 'r', encoding='utf-8') as lib:
25+
data = json.load(lib)
26+
27+
for book in data['books']:
28+
strData = (str(book['id'])+ ' ' + book['title'] + ' ' +
29+
book['author'] + ' ' + book['publisher'] + ' ' +
30+
str(book['year']))
31+
qr = gen.makeQR(strData)
32+
print(strData)
33+
qr.save(args.out + '/' + str(book['id']) + '.png')
34+
else:
35+
print('File or directory not exists!')
36+
37+
if __name__ == '__main__':
38+
sys.exit(main() or 0)

0 commit comments

Comments
 (0)