@@ -32,7 +32,6 @@ void init_instance_segmentation(nb::module_& m) {
3232 ov_any_config[item.first ] = pyutils::py_object_to_any (item.second , item.first );
3333 }
3434
35-
3635 return MaskRCNNModel::create_model (model_path, ov_any_config, preload, device);
3736 },
3837 nb::arg (" model_path" ),
@@ -74,76 +73,76 @@ void init_instance_segmentation(nb::module_& m) {
7473 },
7574 nb::rv_policy::reference_internal)
7675 .def_prop_ro (" label_names" ,
77- [](InstanceSegmentationResult& r) {
78- size_t labels_count = static_cast <size_t >(r.segmentedObjects .size ());
79- std::vector<std::string> labels (labels_count);
76+ [](InstanceSegmentationResult& r) {
77+ size_t labels_count = static_cast <size_t >(r.segmentedObjects .size ());
78+ std::vector<std::string> labels (labels_count);
8079
81- for (size_t i = 0 ; i < labels_count; ++i) {
82- labels[i] = r.segmentedObjects [i].label ;
83- }
80+ for (size_t i = 0 ; i < labels_count; ++i) {
81+ labels[i] = r.segmentedObjects [i].label ;
82+ }
8483
85- return labels;
86- })
84+ return labels;
85+ })
8786 .def_prop_ro (" labels" ,
88- [](InstanceSegmentationResult& r) {
89- size_t labels_count = static_cast <size_t >(r.segmentedObjects .size ());
90- std::vector<size_t > labels (labels_count);
87+ [](InstanceSegmentationResult& r) {
88+ size_t labels_count = static_cast <size_t >(r.segmentedObjects .size ());
89+ std::vector<size_t > labels (labels_count);
9190
92- for (size_t i = 0 ; i < labels_count; ++i) {
93- labels[i] = r.segmentedObjects [i].labelID ;
94- }
91+ for (size_t i = 0 ; i < labels_count; ++i) {
92+ labels[i] = r.segmentedObjects [i].labelID ;
93+ }
9594
96- return LabelsOutput (labels.data (), {labels_count}).cast ();
97- })
95+ return LabelsOutput (labels.data (), {labels_count}).cast ();
96+ })
9897 .def_prop_ro (" scores" ,
99- [](InstanceSegmentationResult& r) {
100- size_t scores_count = static_cast <size_t >(r.segmentedObjects .size ());
101- std::vector<float > scores (scores_count);
98+ [](InstanceSegmentationResult& r) {
99+ size_t scores_count = static_cast <size_t >(r.segmentedObjects .size ());
100+ std::vector<float > scores (scores_count);
102101
103- for (size_t i = 0 ; i < scores_count; ++i) {
104- scores[i] = r.segmentedObjects [i].confidence ;
105- }
102+ for (size_t i = 0 ; i < scores_count; ++i) {
103+ scores[i] = r.segmentedObjects [i].confidence ;
104+ }
106105
107- return ScoresOutput (scores.data (), {scores_count}).cast ();
108- })
106+ return ScoresOutput (scores.data (), {scores_count}).cast ();
107+ })
109108 .def_prop_ro (" bboxes" ,
110- [](InstanceSegmentationResult& r) {
111- size_t boxes_count = static_cast <size_t >(r.segmentedObjects .size ());
112- std::vector<std::vector<int >> boxes (boxes_count);
113-
114- for (size_t i = 0 ; i < boxes_count; ++i) {
115- std::vector<int > box (4 );
116- box[0 ] = r.segmentedObjects [i].tl ().x ;
117- box[1 ] = r.segmentedObjects [i].tl ().y ;
118- box[2 ] = r.segmentedObjects [i].br ().x ;
119- box[3 ] = r.segmentedObjects [i].br ().y ;
120- boxes[i] = box;
121- }
122-
123- return boxes;
124- })
109+ [](InstanceSegmentationResult& r) {
110+ size_t boxes_count = static_cast <size_t >(r.segmentedObjects .size ());
111+ std::vector<std::vector<int >> boxes (boxes_count);
112+
113+ for (size_t i = 0 ; i < boxes_count; ++i) {
114+ std::vector<int > box (4 );
115+ box[0 ] = r.segmentedObjects [i].tl ().x ;
116+ box[1 ] = r.segmentedObjects [i].tl ().y ;
117+ box[2 ] = r.segmentedObjects [i].br ().x ;
118+ box[3 ] = r.segmentedObjects [i].br ().y ;
119+ boxes[i] = box;
120+ }
121+
122+ return boxes;
123+ })
125124 .def_prop_ro (" masks" ,
126- [](InstanceSegmentationResult& r) {
127- size_t elements_count = static_cast <size_t >(r.segmentedObjects .size ());
128- std::vector<std::vector<std::vector<int >>> masks (elements_count);
125+ [](InstanceSegmentationResult& r) {
126+ size_t elements_count = static_cast <size_t >(r.segmentedObjects .size ());
127+ std::vector<std::vector<std::vector<int >>> masks (elements_count);
129128
130- for (size_t i = 0 ; i < elements_count; ++i) {
131- int rows = r.segmentedObjects [i].mask .rows ;
132- int cols = r.segmentedObjects [i].mask .cols ;
129+ for (size_t i = 0 ; i < elements_count; ++i) {
130+ int rows = r.segmentedObjects [i].mask .rows ;
131+ int cols = r.segmentedObjects [i].mask .cols ;
133132
134- std::vector<std::vector<int >> mask (rows, std::vector<int >(cols));
133+ std::vector<std::vector<int >> mask (rows, std::vector<int >(cols));
135134
136- for (int row = 0 ; row < rows; ++row) {
137- for (int col = 0 ; col < cols; ++col) {
138- mask[row][col] = r.segmentedObjects [i].mask .at <uint8_t >(row, col);
139- }
140- }
135+ for (int row = 0 ; row < rows; ++row) {
136+ for (int col = 0 ; col < cols; ++col) {
137+ mask[row][col] = r.segmentedObjects [i].mask .at <uint8_t >(row, col);
138+ }
139+ }
141140
142- masks[i] = mask;
143- }
141+ masks[i] = mask;
142+ }
144143
145- return masks;
146- })
144+ return masks;
145+ })
147146 .def_prop_ro (
148147 " saliency_map" ,
149148 [](InstanceSegmentationResult& r) {
@@ -154,10 +153,9 @@ void init_instance_segmentation(nb::module_& m) {
154153 int cols = r.saliency_map [0 ].cols ;
155154 int num_matrices = r.saliency_map .size ();
156155
157- return nb::ndarray<uint8_t , nb::numpy, nb::c_contig>(&r.saliency_map ,
158- {static_cast <size_t >(num_matrices),
159- static_cast <size_t >(rows),
160- static_cast <size_t >(cols)});
156+ return nb::ndarray<uint8_t , nb::numpy, nb::c_contig>(
157+ &r.saliency_map ,
158+ {static_cast <size_t >(num_matrices), static_cast <size_t >(rows), static_cast <size_t >(cols)});
161159 },
162160 nb::rv_policy::reference_internal);
163161}
0 commit comments