#include "utils.h" #include #include #include bool isImagePath(const std::filesystem::path& path) { return std::filesystem::is_regular_file(path) && (path.extension() == ".png" || path.extension() == ".jpg" || path.extension() == ".jpeg"); } void getImageFiles(const std::filesystem::path& path, std::vector& paths) { if(isImagePath(path)) { paths.push_back(path); } else if(std::filesystem::is_directory(path)) { for(const std::filesystem::directory_entry& dirent : std::filesystem::directory_iterator(path)) { if(std::filesystem::is_directory(dirent.path())) getImageFiles(dirent.path(), paths); else if(isImagePath(dirent.path())) paths.push_back(dirent.path()); } } } cv::Rect rectFromPoints(const std::vector>& points) { int left = std::numeric_limits::max(); int right = std::numeric_limits::min(); int top = std::numeric_limits::max(); int bottom = std::numeric_limits::min(); for(const std::pair& point : points) { left = point.first.x < left ? point.first.x : left; right = point.first.x > right ? point.first.x : right; top = point.first.y < top ? point.first.y : top; bottom = point.first.y > bottom ? point.first.y : bottom; } return cv::Rect(left, top, right-left, bottom-top); } double pointDist(const cv::Point2i& pointA, const cv::Point2i& pointB) { cv::Vec2i a(pointA.x, pointA.y); cv::Vec2i b(pointB.x, pointB.y); return cv::norm(a-b); } bool pointInRect(const cv::Point2i& point, const cv::Rect& rect) { return point.x >= rect.x && point.x <= rect.x+rect.width && point.y >= rect.y && point.y <= rect.y+rect.height; }