changes

scripts/welspun_classifier.py

@@ -228,7 +228,11 @@ class Welspun_Classifier(ModelWrapper):
 
         if a == 0 and x[0] > 0.95:
             if self.defect_type == 'Mix':
-                pass
+                cv2.putText(frame, text="Mix Color Defect Detected", org=(50, 50),
+                            color=(0, 0, 255),
+                            thickness=2,
+                            fontScale=1, fontFace=cv2.LINE_AA)
+                # pass
             else:
                 self.defect_type = 'Mix'
                 resized_frame = cv2.resize(frame, (64, 64))
@@ -248,7 +252,11 @@ class Welspun_Classifier(ModelWrapper):
                 #     self.counter = 0
         elif a == 2 and x[2] > 0.95:
             if self.defect_type == 'Short':
-                pass
+                cv2.putText(frame, text="Short Defect Detected", org=(50, 50),
+                            color=(0, 0, 255),
+                            thickness=2,
+                            fontScale=1, fontFace=cv2.LINE_AA)
+                # pass
             else:
                 self.defect_type = 'Short'
                 resized_frame = cv2.resize(frame, (64, 64))
@@ -268,7 +276,11 @@ class Welspun_Classifier(ModelWrapper):
             #     self.counter = 0
         elif a == 3 and x[3] > 0.95:
             if self.defect_type == 'Split':
-                pass
+                cv2.putText(frame, text="Split Defect Detected", org=(50, 50),
+                            color=(0, 0, 255),
+                            thickness=2,
+                            fontScale=1, fontFace=cv2.LINE_AA)
+                # pass
             else:
                 self.defect_type = 'Split'
                 resized_frame = cv2.resize(frame, (64, 64))