Skip to content

W
Welspun-Classification
Commit 96194fcf authored by dasharatha.vamshi's avatar dasharatha.vamshi
Browse files
Options

openvino

parent 25b9f75a
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
scripts/welspun_classifier.py
View file @ 96194fcf
...@@ -220,6 +220,7 @@ class Welspun_Classifier(ModelWrapper): ...@@ -220,6 +220,7 @@ class Welspun_Classifier(ModelWrapper):
x.append(prob[0]) x.append(prob[0])
x.append(prob[1]) x.append(prob[1])
x.append(prob[2]) x.append(prob[2])
x.append(prob[3])
a = x.index(max(x)) a = x.index(max(x))
# print(type(prob)) # print(type(prob))
# if self.counter%1 == 0: # if self.counter%1 == 0:
...@@ -231,7 +232,7 @@ class Welspun_Classifier(ModelWrapper): ...@@ -231,7 +232,7 @@ class Welspun_Classifier(ModelWrapper):
color=(0, 0, 255), color=(0, 0, 255),
thickness=2, thickness=2,
fontScale=1, fontFace=cv2.LINE_AA) fontScale=1, fontFace=cv2.LINE_AA)
if self.counter % 25 == 0: if self.counter % 5 == 0:
self.send_payload("Mix Color Detected", resized_frame, "Mix Color " + str(prob[0]), "#472020", self.send_payload("Mix Color Detected", resized_frame, "Mix Color " + str(prob[0]), "#472020",
"#ed2020", "#ed2020",
"sound_1") "sound_1")
...@@ -243,7 +244,7 @@ class Welspun_Classifier(ModelWrapper): ...@@ -243,7 +244,7 @@ class Welspun_Classifier(ModelWrapper):
color=(0, 0, 255), color=(0, 0, 255),
thickness=2, thickness=2,
fontScale=1, fontFace=cv2.LINE_AA) fontScale=1, fontFace=cv2.LINE_AA)
if self.counter % 25 == 0: if self.counter % 5 == 0:
self.send_payload("Short Tile Detected", resized_frame, "Short Tile " + str(prob[2]), "#472020", self.send_payload("Short Tile Detected", resized_frame, "Short Tile " + str(prob[2]), "#472020",
"#ed2020", "#ed2020",
"sound_1") "sound_1")
...@@ -255,7 +256,7 @@ class Welspun_Classifier(ModelWrapper): ...@@ -255,7 +256,7 @@ class Welspun_Classifier(ModelWrapper):
color=(0, 0, 255), color=(0, 0, 255),
thickness=2, thickness=2,
fontScale=1, fontFace=cv2.LINE_AA) fontScale=1, fontFace=cv2.LINE_AA)
if self.counter % 25 == 0: if self.counter % 5 == 0:
self.send_payload("Split Defect Detected", resized_frame, "Split " + str(prob[3]), "#472020", "#ed2020", self.send_payload("Split Defect Detected", resized_frame, "Split " + str(prob[3]), "#472020", "#ed2020",
"sound_1") "sound_1")
logger.info(f"Probability: {prob}") logger.info(f"Probability: {prob}")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment