Skip to content

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

good sample to no defect

parent dc572c36
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
scripts/welspun_classifier.py
View file @ 25b9f75a
...@@ -43,7 +43,7 @@ class Welspun_Classifier(ModelWrapper): ...@@ -43,7 +43,7 @@ class Welspun_Classifier(ModelWrapper):
# open-vino # open-vino
self.sink_layer = {'0': 'conv2d_58/BiasAdd/Add', '1': 'conv2d_66/BiasAdd/Add', '2': 'conv2d_74/BiasAdd/Add'} self.sink_layer = {'0': 'conv2d_58/BiasAdd/Add', '1': 'conv2d_66/BiasAdd/Add', '2': 'conv2d_74/BiasAdd/Add'}
self.model_detector_pth = os.path.join(self.base_model_path, "resnet34-wel.xml") self.model_detector_pth = os.path.join(self.base_model_path, "resnet34-wel4.xml")
self.model_bin = os.path.splitext(self.model_detector_pth)[0] + ".bin" self.model_bin = os.path.splitext(self.model_detector_pth)[0] + ".bin"
self.ie = IECore() self.ie = IECore()
self.net1 = self.ie.read_network(model=self.model_detector_pth, weights=self.model_bin) self.net1 = self.ie.read_network(model=self.model_detector_pth, weights=self.model_bin)
......
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