Skip to content

Commit 432b0f8

Browse files
authored
AC: small fixes for custom evaluator (#3228)
1 parent c0bdef9 commit 432b0f8

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/annotation_converters/smartlab_action_recognition.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,8 @@ def convert(self, check_content=False, progress_callback=None, progress_interval
4343
num_iterations = len(stream_description)
4444
for idx, annotation_line in enumerate(stream_description):
4545
identifier, label = annotation_line.split()
46-
annotations.append(ClassificationAnnotation(identifier, int(label)))
46+
label = int(label) if int(label) in [1, 2] else 0
47+
annotations.append(ClassificationAnnotation(identifier, label))
4748
if progress_callback and idx % progress_interval == 0:
4849
progress_callback(idx * 100 / num_iterations)
4950
return ConverterReturn(annotations, self.get_meta(), None)

tools/accuracy_checker/openvino/tools/accuracy_checker/evaluators/custom_evaluators/base_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -306,7 +306,7 @@ def load_model(self, network_info, launcher, log=False):
306306
if weights:
307307
self.network = launcher.read_network(str(model), str(weights))
308308
else:
309-
self.network = launcher.ie_core.read_network(str(model))
309+
self.network = launcher.read_network(str(model), None)
310310
self.load_network(self.network, launcher)
311311
self.set_input_and_output()
312312
if log:

0 commit comments

Comments
 (0)