mutarisi commited on
Commit
812faa8
·
1 Parent(s): ac660bb

updated controller

Browse files
Files changed (1) hide show
  1. lettersControllerS.py +10 -5
lettersControllerS.py CHANGED
@@ -151,14 +151,19 @@ async def detectFromImageBytes(sequenceBytesList, websocket: WebSocket = None, i
151
  return {'letter': '', 'confidenceLetter': 0.0, 'number': '', 'confidenceNumber': 0.0}
152
 
153
  elif numFrames >= 10 and isDynamic:
 
154
  label2, confidence2 = processSequence(sequenceBytesList[:10])
155
  if label2 is None:
156
  return {'letter': '', 'confidenceLetter': 0.0, 'number': '', 'confidenceNumber': 0.0}
157
  _, _, label3, confidence3 = processSingleFrame(sequenceBytesList[-1])
158
  if confidence2 >= 0.6:
159
- return {'letter': label2, 'confidenceLetter': confidence2,
 
 
 
 
 
 
 
160
  'number': label3, 'confidenceNumber': confidence3}
161
- return {'letter': '', 'confidenceLetter': 0.0, 'number': label3, 'confidenceNumber': confidence3}
162
-
163
- else:
164
- return {'letter': '', 'confidenceLetter': 0.0, 'number': '', 'confidenceNumber': 0.0}
 
151
  return {'letter': '', 'confidenceLetter': 0.0, 'number': '', 'confidenceNumber': 0.0}
152
 
153
  elif numFrames >= 10 and isDynamic:
154
+ label1, confidence1, _, _ = processSingleFrame(sequenceBytesList[0])
155
  label2, confidence2 = processSequence(sequenceBytesList[:10])
156
  if label2 is None:
157
  return {'letter': '', 'confidenceLetter': 0.0, 'number': '', 'confidenceNumber': 0.0}
158
  _, _, label3, confidence3 = processSingleFrame(sequenceBytesList[-1])
159
  if confidence2 >= 0.6:
160
+ if label1 == 'I':
161
+ if label2 == 'J':
162
+ return {'letter': label2, 'confidenceLetter': confidence2,
163
+ 'number': label3, 'confidenceNumber': confidence3}
164
+ else:
165
+ return {'letter': label1, 'confidenceLetter': confidence1,
166
+ 'number': label3, 'confidenceNumber': confidence3}
167
+ return {'letter': label1, 'confidenceLetter': confidence1,
168
  'number': label3, 'confidenceNumber': confidence3}
169
+ return {'letter': '', 'confidenceLetter': 0.0, 'number': label3, 'confidenceNumber': confidence3}