ncnn mat pixel affine family

Author: nihui

CMakeLists.txt

@@ -42,6 +42,7 @@ option(NCNN_THREADS "build with threads" ON)
 option(NCNN_BENCHMARK "print benchmark information for every layer" OFF)
 option(NCNN_PIXEL "convert and resize from/to image pixel" ON)
 option(NCNN_PIXEL_ROTATE "rotate image pixel orientation" ON)
+option(NCNN_PIXEL_AFFINE "warp affine image pixel" ON)
 option(NCNN_CMAKE_VERBOSE "print verbose cmake messages" OFF)
 option(NCNN_VULKAN "vulkan compute support" OFF)
 option(NCNN_VULKAN_ONLINE_SPIRV "online SPIR-V module compilation" ON)
@@ -83,7 +84,9 @@ elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^(powerpc)")
     set(NCNN_TARGET_ARCH powerpc)
 else()
     set(NCNN_TARGET_ARCH x86)
-    option(NCNN_AVX2 "optimize x86 platform with avx2" ON)
+    if(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
+        option(NCNN_AVX2 "optimize x86 platform with avx2" ON)
+    endif()
 endif()
 
 message(STATUS "Target arch: ${NCNN_TARGET_ARCH}")

src/CMakeLists.txt

@@ -31,6 +31,7 @@ set(ncnn_SRCS
     layer.cpp
     mat.cpp
     mat_pixel.cpp
+    mat_pixel_affine.cpp
     mat_pixel_resize.cpp
     mat_pixel_rotate.cpp
     modelbin.cpp

src/mat.h

@@ -571,6 +571,26 @@ void kanna_rotate_c4(const unsigned char* src, int srcw, int srch, int srcstride
 // image pixel kanna rotate, convenient wrapper for yuv420sp(nv21/nv12)
 void kanna_rotate_yuv420sp(const unsigned char* src, int srcw, int srch, unsigned char* dst, int w, int h, int type);
 #endif // NCNN_PIXEL_ROTATE
+#if NCNN_PIXEL_AFFINE
+// resolve affine transform matrix from rotation angle, scale factor and x y offset
+void get_rotation_matrix(float angle, float scale, float dx, float dy, float* tm);
+// resolve affine transform matrix from two set of points, num_point must be >= 2
+void get_affine_transform(const float* points_from, const float* points_to, int num_point, float* tm);
+// resolve the inversion affine transform matrix
+void invert_affine_transform(const float* tm, float* tm_inv);
+// image pixel bilinear warpaffine, set -233 for transparent border color
+void warpaffine_bilinear_c1(const unsigned char* src, int srcw, int srch, unsigned char* dst, int w, int h, const float* tm, int v = 0);
+void warpaffine_bilinear_c2(const unsigned char* src, int srcw, int srch, unsigned char* dst, int w, int h, const float* tm, int v = 0);
+void warpaffine_bilinear_c3(const unsigned char* src, int srcw, int srch, unsigned char* dst, int w, int h, const float* tm, int v = 0);
+void warpaffine_bilinear_c4(const unsigned char* src, int srcw, int srch, unsigned char* dst, int w, int h, const float* tm, int v = 0);
+// image pixel bilinear warpaffine with stride(bytes-per-row) parameter, set -233 for transparent border color
+void warpaffine_bilinear_c1(const unsigned char* src, int srcw, int srch, int srcstride, unsigned char* dst, int w, int h, int stride, const float* tm, int v = 0);
+void warpaffine_bilinear_c2(const unsigned char* src, int srcw, int srch, int srcstride, unsigned char* dst, int w, int h, int stride, const float* tm, int v = 0);
+void warpaffine_bilinear_c3(const unsigned char* src, int srcw, int srch, int srcstride, unsigned char* dst, int w, int h, int stride, const float* tm, int v = 0);
+void warpaffine_bilinear_c4(const unsigned char* src, int srcw, int srch, int srcstride, unsigned char* dst, int w, int h, int stride, const float* tm, int v = 0);
+// image pixel bilinear warpaffine, convenient wrapper for yuv420sp(nv21/nv12), set -233 for transparent border color
+void warpaffine_bilinear_yuv420sp(const unsigned char* src, int srcw, int srch, unsigned char* dst, int w, int h, const float* tm, int v = 0);
+#endif // NCNN_PIXEL_AFFINE
 
 // type conversion
 // convert float to half precision floating point