Skip to content

Commit f40a10d

Browse files
committed
Formatting
1 parent 87f9aa4 commit f40a10d

File tree

4 files changed

+33
-35
lines changed

4 files changed

+33
-35
lines changed

src/cpp/include/tasks/instance_segmentation.h

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ class InstanceSegmentation {
1818
std::shared_ptr<InferenceAdapter> adapter;
1919
std::unique_ptr<Pipeline<InstanceSegmentationResult>> pipeline;
2020

21-
InstanceSegmentation(std::shared_ptr<InferenceAdapter> adapter, const ov::AnyMap& configuration) : adapter(adapter) {
21+
InstanceSegmentation(std::shared_ptr<InferenceAdapter> adapter, const ov::AnyMap& configuration)
22+
: adapter(adapter) {
2223
auto config = adapter->getModelConfig();
2324
tiling = utils::get_from_any_maps("tiling", configuration, config, tiling);
2425
if (tiling) {
@@ -75,14 +76,12 @@ class InstanceSegmentation {
7576
const std::vector<cv::Rect>&,
7677
const utils::TilingInfo&);
7778

78-
7979
static std::vector<SegmentedObjectWithRects> getRotatedRectangles(const InstanceSegmentationResult& result);
8080
static std::vector<Contour> getContours(const std::vector<SegmentedObject>& objects);
8181

8282
bool postprocess_semantic_masks = true;
8383

8484
private:
85-
8685
bool tiling;
8786

8887
utils::RESIZE_MODE resize_mode;

src/cpp/include/utils/config.h

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ enum RESIZE_MODE {
1616
NO_RESIZE,
1717
};
1818

19-
2019
template <typename Type>
2120
Type get_from_any_maps(const std::string& key,
2221
const ov::AnyMap& top_priority,
@@ -51,12 +50,11 @@ inline bool get_from_any_maps(const std::string& key,
5150
return low_priority;
5251
}
5352

54-
template<>
53+
template <>
5554
inline RESIZE_MODE get_from_any_maps(const std::string& key,
56-
const ov::AnyMap& top_priority,
57-
const ov::AnyMap& mid_priority,
58-
RESIZE_MODE low_priority) {
59-
55+
const ov::AnyMap& top_priority,
56+
const ov::AnyMap& mid_priority,
57+
RESIZE_MODE low_priority) {
6058
std::string resize_type = "standard";
6159
resize_type = utils::get_from_any_maps("resize_type", top_priority, mid_priority, resize_type);
6260
RESIZE_MODE resize = RESIZE_FILL;
@@ -75,8 +73,6 @@ inline RESIZE_MODE get_from_any_maps(const std::string& key,
7573
return resize;
7674
}
7775

78-
79-
8076
inline bool model_has_embedded_processing(std::shared_ptr<ov::Model> model) {
8177
if (model->has_rt_info("model_info")) {
8278
auto model_info = model->get_rt_info<ov::AnyMap>("model_info");

src/cpp/include/utils/preprocessing.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
#include <opencv2/opencv.hpp>
99
#include <openvino/openvino.hpp>
10+
1011
#include "utils/config.h"
1112

1213
namespace utils {

src/cpp/src/tasks/instance_segmentation.cpp

Lines changed: 26 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@
88
#include "adapters/openvino_adapter.h"
99
#include "utils/config.h"
1010
#include "utils/math.h"
11+
#include "utils/nms.h"
1112
#include "utils/preprocessing.h"
1213
#include "utils/tensor.h"
13-
#include "utils/nms.h"
1414

1515
constexpr char saliency_map_name[]{"saliency_map"};
1616
constexpr char feature_vector_name[]{"feature_vector"};
@@ -232,7 +232,7 @@ InstanceSegmentationResult InstanceSegmentation::postprocess(InferenceResult& in
232232
std::cout << "resize mode: " << resize_mode << std::endl;
233233
int padLeft = 0, padTop = 0;
234234
if (utils::RESIZE_KEEP_ASPECT == resize_mode || utils::RESIZE_KEEP_ASPECT_LETTERBOX == resize_mode) {
235-
std::cout << "using some other resize mode..." << std::endl;
235+
std::cout << "using some other resize mode..." << std::endl;
236236
invertedScaleX = invertedScaleY = std::max(invertedScaleX, invertedScaleY);
237237
if (utils::RESIZE_KEEP_ASPECT_LETTERBOX == resize_mode) {
238238
padLeft = (input_shape.width - int(std::round(floatInputImgWidth / invertedScaleX))) / 2;
@@ -306,7 +306,8 @@ InstanceSegmentationResult InstanceSegmentation::postprocess(InferenceResult& in
306306
return result;
307307
}
308308

309-
InstanceSegmentationResult InstanceSegmentation::postprocess_tile(InstanceSegmentationResult result, const cv::Rect& coord) {
309+
InstanceSegmentationResult InstanceSegmentation::postprocess_tile(InstanceSegmentationResult result,
310+
const cv::Rect& coord) {
310311
for (auto& det : result.segmentedObjects) {
311312
det.x += coord.x;
312313
det.y += coord.y;
@@ -322,11 +323,12 @@ InstanceSegmentationResult InstanceSegmentation::postprocess_tile(InstanceSegmen
322323
return result;
323324
}
324325

325-
InstanceSegmentationResult InstanceSegmentation::merge_tiling_results(const std::vector<InstanceSegmentationResult>& tiles_results,
326-
const cv::Size& image_size,
327-
const std::vector<cv::Rect>& tile_coords,
328-
const utils::TilingInfo& tiling_info) {
329-
size_t max_pred_number = 200; //TODO: Actually get this from config!
326+
InstanceSegmentationResult InstanceSegmentation::merge_tiling_results(
327+
const std::vector<InstanceSegmentationResult>& tiles_results,
328+
const cv::Size& image_size,
329+
const std::vector<cv::Rect>& tile_coords,
330+
const utils::TilingInfo& tiling_info) {
331+
size_t max_pred_number = 200; // TODO: Actually get this from config!
330332

331333
InstanceSegmentationResult output;
332334
std::vector<AnchorLabeled> all_detections;
@@ -346,15 +348,15 @@ InstanceSegmentationResult InstanceSegmentation::merge_tiling_results(const std:
346348
output.segmentedObjects.reserve(keep_idx.size());
347349
for (auto idx : keep_idx) {
348350
if (postprocess_semantic_masks) {
349-
//why does this happen again?
350-
//all_detections_ptrs[idx].get().mask = ;
351-
//SegmentedObject obj = all_detections_ptrs[idx]; //copy
352-
//std::cout << "Mask size before: " << obj.mask.size() << std::endl;
353-
//std::cout << static_cast<cv::Rect>(obj) << std::endl;
354-
//obj.mask = segm_postprocess(all_detections_ptrs[idx],
355-
// obj.mask,
356-
// image_size.height,
357-
// image_size.width);
351+
// why does this happen again?
352+
// all_detections_ptrs[idx].get().mask = ;
353+
// SegmentedObject obj = all_detections_ptrs[idx]; //copy
354+
// std::cout << "Mask size before: " << obj.mask.size() << std::endl;
355+
// std::cout << static_cast<cv::Rect>(obj) << std::endl;
356+
// obj.mask = segm_postprocess(all_detections_ptrs[idx],
357+
// obj.mask,
358+
// image_size.height,
359+
// image_size.width);
358360
}
359361

360362
output.segmentedObjects.push_back(all_detections_ptrs[idx]);
@@ -390,14 +392,13 @@ InstanceSegmentationResult InstanceSegmentation::merge_tiling_results(const std:
390392
output.saliency_map = merge_saliency_maps(tiles_results, image_size, tile_coords, tiling_info);
391393

392394
return output;
393-
394395
}
395396

396-
397-
std::vector<cv::Mat_<std::uint8_t>> InstanceSegmentation::merge_saliency_maps(const std::vector<InstanceSegmentationResult>& tiles_results,
398-
const cv::Size& image_size,
399-
const std::vector<cv::Rect>& tile_coords,
400-
const utils::TilingInfo& tiling_info ) {
397+
std::vector<cv::Mat_<std::uint8_t>> InstanceSegmentation::merge_saliency_maps(
398+
const std::vector<InstanceSegmentationResult>& tiles_results,
399+
const cv::Size& image_size,
400+
const std::vector<cv::Rect>& tile_coords,
401+
const utils::TilingInfo& tiling_info) {
401402
std::vector<std::vector<cv::Mat_<std::uint8_t>>> all_saliency_maps;
402403
all_saliency_maps.reserve(tiles_results.size());
403404
for (const auto& result : tiles_results) {
@@ -435,7 +436,8 @@ std::vector<cv::Mat_<std::uint8_t>> InstanceSegmentation::merge_saliency_maps(co
435436
}
436437

437438
for (size_t class_idx = 0; class_idx < num_classes; ++class_idx) {
438-
auto image_map_cls = tiling_info.tile_with_full_image ? image_saliency_map[class_idx] : cv::Mat_<std::uint8_t>();
439+
auto image_map_cls =
440+
tiling_info.tile_with_full_image ? image_saliency_map[class_idx] : cv::Mat_<std::uint8_t>();
439441
if (image_map_cls.empty()) {
440442
if (cv::sum(merged_map[class_idx]) == cv::Scalar(0.)) {
441443
merged_map[class_idx] = cv::Mat_<std::uint8_t>();

0 commit comments

Comments
 (0)