Skip to content

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

changes

parent 525d15c1
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 4735 additions and 1 deletion
scripts/model/resnet34-demo1.mapping 0 → 100644
View file @ 35d5abe5
This diff is collapsed.
scripts/model/resnet34-demo1.xml 0 → 100644
View file @ 35d5abe5
This source diff could not be displayed because it is too large. You can view the blob instead.
scripts/welspun_classifier.py
View file @ 35d5abe5
...@@ -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-demo.xml") self.model_detector_pth = os.path.join(self.base_model_path, "resnet34-demo1.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