Spaces:
Sleeping
Sleeping
mutarisi
commited on
Commit
·
812faa8
1
Parent(s):
ac660bb
updated controller
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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}
|
|
|
|
|
|
|
|
|