assembly-opencv 1.0.0__py3-none-any.whl

assembly/PolygonOperation.py

@@ -0,0 +1,323 @@
+import cv2
+import numpy as np
+from shapely import Polygon
+
+from assembly.description.Decoration import func_description, class_description
+from assembly.GeneralUtil import show_image
+
+'''
+轮廓运算
+
+输入：轮廓矩阵
+输出：轮廓矩阵
+'''
+
+
+@class_description(
+    title="轮廓运算",
+    description="轮廓运算主要是指对轮廓进行各种数学运算以实现轮廓的增强、变换和分析等功能。",
+    level=10
+)
+class PolygonOperation(object):
+
+    @func_description(
+        title="轮廓平移",
+        description="对轮廓进行平移操作"
+    )
+    @staticmethod
+    def polygon_translate(contour, tx, ty):
+        """
+        :name 轮廓平移
+        :param contour: 轮廓矩阵
+        :param tx: x方向平移距离
+        :param ty: y方向平移距离
+        :return contour: 平移后的轮廓矩阵
+        """
+        M = np.array([[1, 0, tx], [0, 1, ty]], dtype=np.float32)
+        contour_translated = cv2.transform(contour, M)
+        return contour_translated
+
+    @func_description(
+        title="轮廓旋转",
+        description="对轮廓进行旋转操作"
+    )  
+    @staticmethod
+    def polygon_rotate(contour, angle, center=None):
+        """
+        :name 轮廓旋转
+        :param contour: 轮廓矩阵
+        :param angle: 旋转角度，正值表示逆时针旋转，负值表示顺时针旋转
+        :param center: 旋转中心点，默认None表示以轮廓的质心为中心
+        :return contour: 旋转后的轮廓矩阵
+        """
+        if center is None:
+            M = cv2.moments(contour)
+            if M["m00"] != 0:
+                center = (int(M["m10"] / M["m00"]), int(M["m01"] / M["m00"]))
+            else:
+                center = (0, 0)
+
+        M = cv2.getRotationMatrix2D(center, angle, 1.0)
+        contour_rotated = cv2.transform(contour, M)
+        return contour_rotated
+
+    @func_description(
+        title="轮廓缩放",
+        description="对轮廓进行缩放操作"
+    )
+    @staticmethod
+    def polygon_scale(contour, scale_x, scale_y, center=None):
+        """
+        :name 轮廓缩放
+        :param contour: 轮廓矩阵
+        :param scale_x: x方向缩放因子
+        :param scale_y: y方向缩放因子
+        :param center: 缩放中心点，默认None表示以轮廓的质心为中心
+        :return contour: 缩放后的轮廓矩阵
+        """
+        if center is None:
+            M = cv2.moments(contour)
+            if M["m00"] != 0:
+                center = (int(M["m10"] / M["m00"]), int(M["m01"] / M["m00"]))
+            else:
+                center = (0, 0)
+
+        M = np.array([[scale_x, 0, (1 - scale_x) * center[0]],
+                      [0, scale_y, (1 - scale_y) * center[1]]], dtype=np.float32)
+        contour_scaled = cv2.transform(contour, M)
+        return contour_scaled
+
+    @func_description(
+        title="轮廓取交集",
+        description="计算两个轮廓的交集区域"
+    )
+    @staticmethod
+    def contours_intersect(contour1, contour2, buffer=0):
+        """
+        :name 轮廓取交集
+        :param contour1: 轮廓矩阵1
+        :param contour2: 轮廓矩阵2
+        :param buffer: 扩展轮廓 >0外扩 <0内缩
+        :return intersect: 布尔值，表示是否相交
+        """
+        poly1 = Polygon(contour1)
+        if buffer != 0:
+            poly1 = poly1.buffer(buffer)
+
+        poly2 = Polygon(contour2)
+        if buffer != 0:
+            poly2 = poly2.buffer(buffer)
+
+        if not poly1.is_valid:
+            poly1 = poly1.buffer(0)
+        if not poly2.is_valid:
+            poly2 = poly2.buffer(0)
+
+        if poly1.intersects(poly2):
+            intersect_poly = poly1.intersection(poly2)
+            intersect = self.shapely_poly_to_cv(intersect_poly)
+        else:
+            intersect = []
+
+        return intersect
+
+    @func_description(
+        title="轮廓取差集",
+        description="计算两个轮廓的差集区域"
+    )
+    @staticmethod
+    def contours_difference(contour1, contour2, buffer=0):
+        """
+        :name 轮廓取差集
+        :param contour1: 轮廓矩阵1
+        :param contour2: 轮廓矩阵2
+        :param buffer: 扩展轮廓 >0外扩 <0内缩
+        :return difference_contour: 差集轮廓矩阵
+        """
+        poly1 = Polygon(contour1)
+        if buffer != 0:
+            poly1 = poly1.buffer(buffer)
+
+        poly2 = Polygon(contour2)
+        if buffer != 0:
+            poly2 = poly2.buffer(buffer)
+
+        if not poly1.is_valid:
+            poly1 = poly1.buffer(0)
+        if not poly2.is_valid:
+            poly2 = poly2.buffer(0)
+
+        if poly1.intersects(poly2):
+            difference_poly = poly1.difference(poly2)
+            difference = self.shapely_poly_to_cv(difference_poly)
+        else:
+            difference = []
+
+        return difference
+
+    @func_description(
+        title="轮廓取并集",
+        description="计算两个轮廓的并集区域"
+    )
+    @staticmethod
+    def contours_union(contour1, contour2, buffer=0):
+        """
+        :name 轮廓取并集
+        :param contour1: 轮廓矩阵1
+        :param contour2: 轮廓矩阵2
+        :param buffer: 扩展轮廓 >0外扩 <0内缩
+        :return union_contour: 并集轮廓矩阵
+        """
+        poly1 = Polygon(contour1)
+        if buffer != 0:
+            poly1 = poly1.buffer(buffer)
+
+        poly2 = Polygon(contour2)
+        if buffer != 0:
+            poly2 = poly2.buffer(buffer)
+
+        if not poly1.is_valid:
+            poly1 = poly1.buffer(0)
+        if not poly2.is_valid:
+            poly2 = poly2.buffer(0)
+
+        if poly1.intersects(poly2):
+            union_poly = poly1.union(poly2)
+            union = self.shapely_poly_to_cv(union_poly)
+        else:
+            union = []
+
+        return union
+
+    @func_description(
+        title="轮廓交并比",
+        description="计算两个轮廓的交并比"
+    )
+    @staticmethod
+    def contours_iou(contour1, contour2, buffer=0):
+        """
+        :name 轮廓交并比
+        :param contour1: 轮廓矩阵1
+        :param contour2: 轮廓矩阵2
+        :param buffer: 扩展轮廓 >0外扩 <0内缩
+        :return iou: 交并比
+        """
+        poly1 = Polygon(contour1)
+        if buffer != 0:
+            poly1 = poly1.buffer(buffer)
+
+        poly2 = Polygon(contour2)
+        if buffer != 0:
+            poly2 = poly2.buffer(buffer)
+
+        if not poly1.is_valid:
+            poly1 = poly1.buffer(0)
+        if not poly2.is_valid:
+            poly2 = poly2.buffer(0)
+
+        if poly1.intersects(poly2):
+            intersection = poly1.intersection(poly2)
+            intersection_area = intersection.area
+
+            union = poly1.union(poly2)
+
+            return intersection_area / union.area
+        else:
+            return 0
+
+    @func_description(
+        title="轮廓绘制",
+        description="在图像上绘制轮廓"
+    )
+    @staticmethod
+    def draw_contours(img, contours, color=(0, 255, 0), thickness=2):
+        """
+        :name 轮廓绘制
+        :param img: 原始图像
+        :param contours: 轮廓矩阵或轮廓矩阵列表
+        :param color: 轮廓颜色，默认为绿色
+        :param thickness: 轮廓线条粗细，默认为2
+        :return img_with_contours: 绘制轮廓后的图像
+        """
+
+        if img is not None:
+            img_with_contours = img.copy()
+        else:
+            contour_width = contours[:, :, 0].max() - contours[:, :, 0].min() + 10
+            contour_height = contours[:, :, 1].max() - contours[:, :, 1].min() + 10
+            img_with_contours = np.zeros((contour_height, contour_width, 3), dtype=np.uint8)
+        if isinstance(contours, list):
+            cv2.drawContours(img_with_contours, contours, -1, color, thickness)
+        else:
+            cv2.drawContours(img_with_contours, [contours], -1, color, thickness)
+        return img_with_contours
+
+    # OpenCV多边形 -> Shapely多边形
+    @staticmethod
+    def cv_poly_to_shapely(cv_poly):
+        # 将点从OpenCV格式（float32）转换为整数格式
+        points = np.round(cv_poly.reshape(-1, 1, 2)).astype(np.int)
+        return Polygon(points)
+
+    # Shapely多边形 -> OpenCV多边形
+    @staticmethod
+    def shapely_poly_to_cv(shapely_poly):
+        # 提取Shapely多边形的顶点
+        vertices = np.array(shapely_poly.exterior.coords)
+        # 转换为float32类型
+        return vertices.astype(int)
+
+
+# 暴露方法
+def polygon_translate(contour, tx, ty):
+    return PolygonOperation.polygon_translate(contour, tx, ty)
+
+def polygon_rotate(contour, angle, center=None):
+    return PolygonOperation.polygon_rotate(contour, angle, center)
+
+def polygon_scale(contour, scale_factor, center=None):
+    return PolygonOperation.polygon_scale(contour, scale_factor, center)
+
+def contours_intersect(contour1, contour2, buffer=0):
+    return PolygonOperation.contours_intersect(contour1, contour2, buffer)
+
+def contours_union(contour1, contour2, buffer=0):
+    return PolygonOperation.contours_union(contour1, contour2, buffer)
+
+def contours_difference(contour1, contour2, buffer=0):
+    return PolygonOperation.contours_difference(contour1, contour2, buffer)
+
+def contours_iou(contour1, contour2, buffer=0):
+    return PolygonOperation.contours_iou(contour1, contour2, buffer)
+
+def draw_contours(img, contours, color=(0, 255, 0), thickness=2):
+    return PolygonOperation.draw_contours(img, contours, color, thickness)
+
+
+
+if __name__ == '__main__':
+    poly = PolygonOperation()
+    polygon1 = [[100, 100], [200, 100], [200, 200], [100, 200]]
+    polygon2 = [[150, 150], [300, 150], [300, 300], [150, 300]]
+
+    img = np.zeros((400, 400, 3), dtype=np.uint8)
+    img1 = poly.draw_contours(img, np.array(polygon1))
+    img2 = poly.draw_contours(img, np.array(polygon2), color=(255, 0, 0))
+
+    show_image("polygon1", img1)
+    show_image("polygon2", img2)
+
+    contours_intersect = poly.contours_intersect(np.array(polygon1), np.array(polygon2))
+    img_contour = poly.draw_contours(img, np.array(contours_intersect))
+    show_image("contours_intersect", img_contour)
+
+    contours_union = poly.contours_union(np.array(polygon1), np.array(polygon2))
+    img_contour = poly.draw_contours(img, np.array(contours_union))
+    show_image("contours_union", img_contour)
+
+    contours_difference = poly.contours_difference(np.array(polygon1), np.array(polygon2))
+    img_contour = poly.draw_contours(img, np.array(contours_difference))
+    show_image("contours_difference", img_contour)
+
+    iou = poly.contours_iou(np.array(polygon1), np.array(polygon2))
+    print("交并比：", iou)

assembly/__init__.py

@@ -0,0 +1,165 @@
+# assembly包初始化文件
+
+# 导出所有模块的函数
+from .CornerDetection import *
+from .DrawImage import *
+from .EdgeDetection import *
+from .GeneralUtil import *
+from .GeometricChanges import *
+from .ImageEnhance import *
+from .ImageOperation import *
+from .ImageSegmentation import *
+from .MorphologicalChange import *
+from .PolygonOperation import *
+
+__all__ = [
+    # CornerDetection
+    'CornerDetection',
+    'harris_corner_detection',
+    'shi_tomasi_corner_detection',
+    'fast_corner_detection',
+    
+    # DrawImage
+    'DrawImage',
+    'draw_circle',
+    'draw_rectangle',
+    'draw_line',
+    'draw_text',
+    'generate_blank_image',
+    
+    # EdgeDetection
+    'EdgeDetection',
+    'canny_edge_detection',
+    'sobel_edge_detection',
+    'laplacian_edge_detection',
+    
+    # GeneralUtil
+    'GeneralUtil',
+    'color_transform_gray',
+    'color_transform_hsv',
+    'color_transform_lab',
+    'color_transform_yuv',
+    'color_transform_rgb',
+    'image_blend',
+    'normalize_image',
+    'resize_image',
+    'rotate_image',
+    'flip_image',
+    'crop_image',
+    'threshold_image',
+    'adaptive_threshold_image',
+    'histogram_equalization',
+    'clahe_equalization',
+    'gaussian_blur',
+    'median_blur',
+    'bilateral_filter',
+    'sharpen_image',
+    'denoise_image',
+    'calculate_histogram',
+    'calculate_mean',
+    'calculate_std',
+    'calculate_min_max',
+    'calculate_entropy',
+    'calculate_gradient',
+    'calculate_laplacian',
+    'calculate_sobel',
+    'calculate_canny',
+    'calculate_harris',
+    'calculate_shi_tomasi',
+    'calculate_fast',
+    'calculate_orb',
+    'calculate_sift',
+    'calculate_surf',
+    'calculate_akaze',
+    'calculate_brisk',
+    'calculate_kaze',
+    'calculate_mser',
+    'calculate_good_features_to_track',
+    'calculate_corner_sub_pix',
+    'calculate_find_homography',
+    'calculate_perspective_transform',
+    'calculate_warp_perspective',
+    'calculate_affine_transform',
+    'calculate_warp_affine',
+    'calculate_remap',
+    'calculate_resize',
+    'calculate_rotate',
+    'calculate_flip',
+    'calculate_crop',
+    'calculate_threshold',
+    'calculate_adaptive_threshold',
+    'calculate_histogram_equalization',
+    'calculate_clahe_equalization',
+    'calculate_gaussian_blur',
+    'calculate_median_blur',
+    'calculate_bilateral_filter',
+    'calculate_sharpen_image',
+    'calculate_denoise_image',
+    
+    # GeometricChanges
+    'GeometricChanges',
+    'geometric_resize',
+    'geometric_rotate',
+    'geometric_flip',
+    'geometric_crop',
+    'geometric_translate',
+    'geometric_scale',
+    'geometric_affine',
+    'geometric_perspective',
+    
+    # ImageEnhance
+    'ImageEnhance',
+    'enhance_brightness',
+    'enhance_contrast',
+    'enhance_saturation',
+    'enhance_sharpness',
+    'enhance_gamma',
+    'enhance_histogram',
+    'enhance_clahe',
+    
+    # ImageOperation
+    'ImageOperation',
+    'image_add',
+    'image_subtract',
+    'image_multiply',
+    'image_divide',
+    'image_bitwise_and',
+    'image_bitwise_or',
+    'image_bitwise_xor',
+    'image_bitwise_not',
+    'image_mask',
+    
+    # ImageSegmentation
+    'ImageSegmentation',
+    'segmentation_threshold',
+    'segmentation_adaptive_threshold',
+    'segmentation_watershed',
+    'segmentation_grabcut',
+    'segmentation_mean_shift',
+    'segmentation_kmeans',
+    'segmentation_spectral',
+    
+    # MorphologicalChange
+    'MorphologicalChange',
+    'mlc_erode',
+    'mlc_dilate',
+    'mlc_open',
+    'mlc_close',
+    'mlc_gradient',
+    'mlc_tophat',
+    'mlc_blackhat',
+    
+    # PolygonOperation
+    'PolygonOperation',
+    'polygon_contour_detection',
+    'polygon_contour_area',
+    'polygon_contour_perimeter',
+    'polygon_convex_hull',
+    'polygon_bounding_rect',
+    'polygon_min_area_rect',
+    'polygon_min_enclosing_circle',
+    'polygon_approx_poly_dp',
+    'polygon_circle_detection',
+    'polygon_line_detection',
+    'polygon_shape_detection',
+]

assembly_opencv-1.0.0.dist-info/METADATA

@@ -0,0 +1,16 @@
+Metadata-Version: 2.1
+Name: assembly-opencv
+Version: 1.0.0
+Summary: Image processing utilities for Python with OpenCV
+Home-page: 
+Author: Zven
+Author-email: zven@zven.xyz
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.6
+Requires-Dist: numpy
+Requires-Dist: opencv-python
+Requires-Dist: Pillow
+
+A collection of image processing functions for computer vision tasks with OpenCV

assembly_opencv-1.0.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+assembly/CornerDetection.py,sha256=3aa6zilFuU8neTyeJgqTMuVobQRmJKbYyrNgD8O5pYI,4195
+assembly/DrawImage.py,sha256=g9t3OBeX_TcdrLK_nNv0rKpsk-NrTs6ng_sYj5t92Xo,10630
+assembly/EdgeDetection.py,sha256=MTMqf-N1eMMctsyGQnOoGY7qnon5O1O5eRm1OYXd_Z4,7743
+assembly/GeneralUtil.py,sha256=yjLNyeXSkJ3VN09kVOYj__ZxbG2uhzCMqqVK__7SRw4,2746
+assembly/GeometricChanges.py,sha256=v-cq0mW4lWoexTvN1q5fJgsMxB9qNzDOwWoDNtMERqM,6007
+assembly/ImageEnhance.py,sha256=_H_-OsYnrARH8b-8C8mrj4KiOQ6xAgH5wTVBgZibGtA,9263
+assembly/ImageOperation.py,sha256=y68NzACSyIPI-5Bm4_2dPvkAs70fDFcuLVh8_spWKz8,5501
+assembly/ImageSegmentation.py,sha256=8pw-gyEO6FeMqUMbSsB7JL86-_0j9lSLAXMTi7i2HCg,7939
+assembly/MorphologicalChange.py,sha256=XbZyVdkJXKmsWok9JOa8toH1GS-pmXrwdm8R757r_fU,9919
+assembly/PolygonOperation.py,sha256=DLEvWEUFY_Rxeiqy1WxKw3qNOkR8eWbiUfuZy9VVyVI,10915
+assembly/__init__.py,sha256=DYzr16XXFUEKvmyqqsnB5BQispL1MRaGGUxyv62dnik,3951
+assembly_opencv-1.0.0.dist-info/METADATA,sha256=h5kxDVVj8Vb1bDdxSq3MGrIl0UVuuNJOoR5-uvnaPtk,510
+assembly_opencv-1.0.0.dist-info/WHEEL,sha256=hPN0AlP2dZM_3ZJZWP4WooepkmU9wzjGgCLCeFjkHLA,92
+assembly_opencv-1.0.0.dist-info/top_level.txt,sha256=Lz_5VS2mRAR7AduMlKl3vVyAOekcs_JNGew8waVVLas,9
+assembly_opencv-1.0.0.dist-info/RECORD,,

assembly_opencv-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.46.3)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

assembly_opencv-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+assembly