changes

scripts/welspun_classifier.py

@@ -209,7 +209,6 @@ class Welspun_Classifier(ModelWrapper):
         # log.info("Batch size is {}".format(n))
         #
         # log.info("Starting inference in synchronous mode")
-        if self.counter % 3 == 0:
         start = time.time()
         res = self.exec_net.infer(inputs={self.input_blob: images})
         print(f"Inference time: {time.time() - start}")
@@ -227,7 +226,7 @@ class Welspun_Classifier(ModelWrapper):
         a = x.index(max(x))
         # print(type(prob))
         # if self.counter%1 == 0:
-            # self.counter = self.counter + 1
+        self.counter = self.counter + 1
 
         if a == 1 and x[1] > 0.95:
             if self.defect_type == 'Mix':
@@ -347,10 +346,9 @@ class Welspun_Classifier(ModelWrapper):
         #                 fontScale=1, fontFace=cv2.LINE_AA)
         #     self.send_payload("Stitch Detected", frame, "Stitch " + str(prob[0]), "#472020", "#ed2020", "sound_1")
         #     logger.info(f"Probability: {prob}")
-            self.counter = self.counter + 1
+        # self.counter= self.counter + 1
         print("total time taken to process-------------> ", str(time.time() - starttime))
         # logger.info(f"total time taken to process----------------- {time.time()-starttime}")
         # cv2.imshow('res', frame)
-        else:
-            pass
+
         return frame