1.0: init

ImageProcessing-ElectronicPublications · Jul 15, 2023 · 261ff8c · 261ff8c
commit 261ff8c
Show file tree

Hide file tree

Showing 40 changed files with 1,509 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,32 @@
+# Compiled Object files
+*.slo
+*.lo
+*.o
+*.obj
+
+# Precompiled Headers
+*.gch
+*.pch
+
+# Compiled Dynamic libraries
+*.so
+*.dylib
+*.dll
+
+# Fortran module files
+*.mod
+*.smod
+
+# Compiled Static libraries
+*.lai
+*.la
+*.a
+*.lib
+
+# Executables
+*.exe
+*.out
+*.app
+opencv-pixartscale*
+
+
diff --git a/LICENSE b/LICENSE
@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2017 Anton Myronyuk
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
diff --git a/Makefile b/Makefile
@@ -0,0 +1,17 @@
+PROJECT = opencv-pixartscale
+CXX = g++
+OCV = opencv4
+CXXFLAGS = -Wall `pkg-config --cflags $(OCV)`
+LDFLAGS = `pkg-config --libs $(OCV)` -s
+OBJS =  src/buildingsimilaritygraph.o src/cells.o src/cellsreconstruct.o src/combine.o src/curvessearching.o src/depixelizing.o src/heuristics.o src/visualization.o src/main.o
+
+all: $(PROJECT)
+
+$(PROJECT): $(OBJS)
+	$(CXX) $^ $(LDFLAGS) -o $@
+
+%.o: %.cpp
+	$(CXX) $(CXXFLAGS) -c $< -o $@
+
+clean:
+	$(RM) $(OBJS) $(PROJECT)
diff --git a/README.md b/README.md
@@ -0,0 +1,27 @@
+# opencv-pixartscale
+
+Scaling up images in low resolution (pixel art only).  
+Uses [OpenCV](https://opencv.org/).
+
+## Build
+
+Type:
+```shell
+make
+```
+
+## Usage
+
+```shell
+opencv-pixartscale image_in image_out scale
+```
+
+## Example
+
+Input (x20):
+
+![input](examples/input-x20-nn.png)
+
+Output (x20 and reconstructed):
+
+![output](examples/output-x20-pixartscale.png)
diff --git a/examples/00.png b/examples/00.png
diff --git a/examples/01.png b/examples/01.png
diff --git a/examples/02.png b/examples/02.png
diff --git a/examples/03.png b/examples/03.png
diff --git a/examples/04.png b/examples/04.png
diff --git a/examples/05.png b/examples/05.png
diff --git a/examples/06.png b/examples/06.png
diff --git a/examples/10.png b/examples/10.png
diff --git a/examples/11.png b/examples/11.png
diff --git a/examples/12.png b/examples/12.png
diff --git a/examples/13.png b/examples/13.png
diff --git a/examples/14.png b/examples/14.png
diff --git a/examples/15.png b/examples/15.png
diff --git a/examples/16.png b/examples/16.png
diff --git a/examples/17.png b/examples/17.png
diff --git a/examples/18.png b/examples/18.png
diff --git a/examples/19.png b/examples/19.png
diff --git a/examples/20.png b/examples/20.png
diff --git a/examples/21.png b/examples/21.png
diff --git a/examples/22.png b/examples/22.png
diff --git a/examples/333.png b/examples/333.png
diff --git a/examples/input-x20-nn.png b/examples/input-x20-nn.png
diff --git a/examples/mario.png b/examples/mario.png
diff --git a/examples/output-x20-pixartscale.png b/examples/output-x20-pixartscale.png
diff --git a/examples/t.png b/examples/t.png
diff --git a/src/buildingsimilaritygraph.cpp b/src/buildingsimilaritygraph.cpp
@@ -0,0 +1,106 @@
+#include "depixelizing.h"
+void Depixelizing::buildSimilarityGraph()
+{
+    cv::Mat imgYUV(m_image.size(), CV_8UC3);
+    cv::cvtColor(m_image, imgYUV, cv::COLOR_BGR2YUV);
+    m_connections.resize(m_width * m_height);
+    //building
+    for (int i = 0; i < m_height; ++i)
+    {
+        for (int j = 0; j < m_width; ++j)
+        {
+            cv::Vec3b& currentPixel = imgYUV.at<cv::Vec3b>(i, j);
+            //TOP
+            if (i > 0)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i - 1, j);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_TOP;
+                }
+            }
+            //TOP_RIGHT
+            if (j < m_width - 1 && i > 0)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i - 1, j + 1);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_TOP_RIGHT;
+                }
+            }
+            //RIGHT
+            if (j < m_width - 1)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i, j + 1);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_RIGHT;
+                }
+            }
+            //BOTTOM_RIGHT
+            if (j < m_width - 1 && i < m_height - 1)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i + 1, j + 1);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_BOTTOM_RIGHT;
+                }
+            }
+            //BOTTOM
+            if (i < m_height - 1)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i + 1, j);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_BOTTOM;
+                }
+            }
+            //BOTTOM_LEFT
+            if (j > 0 && i < m_height - 1)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i + 1, j - 1);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_BOTTOM_LEFT;
+                }
+            }
+            //LEFT
+            if (j > 0)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i, j - 1);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_LEFT;
+                }
+            }
+            //TOP_LEFT
+            if (j > 0 && i > 0)
+            {
+                cv::Vec3b& neighbourPixel = imgYUV.at<cv::Vec3b>(i - 1, j - 1);
+                if (abs(currentPixel[0] - neighbourPixel[0]) < 48 && abs(currentPixel[1] - neighbourPixel[1]) < 7 && abs(currentPixel[2] - neighbourPixel[2]) < 6)
+                {
+                    m_connections[i * m_width + j] |= NEIGHBOUR_TOP_LEFT;
+                }
+            }
+        }
+    }
+}
+void Depixelizing::deleteFullyConnectedBlocks()
+{
+    for (int i = 0; i < m_height - 1; ++i)
+    {
+        for (int j = 0; j < m_width - 1; ++j)
+        {
+            if ((m_connections[i * m_width + j] & NEIGHBOUR_RIGHT) &&
+                    (m_connections[i * m_width + j] & NEIGHBOUR_BOTTOM) &&
+                    (m_connections[(i + 1) * m_width + j] & NEIGHBOUR_RIGHT) &&
+                    (m_connections[i * m_width + (j + 1)] & NEIGHBOUR_BOTTOM))
+            {
+                m_connections[i * m_width + j] &= ~NEIGHBOUR_BOTTOM_RIGHT;
+                m_connections[(i + 1) * m_width + (j + 1)] &= ~NEIGHBOUR_TOP_LEFT;
+                m_connections[(i + 1) * m_width + j] &= ~NEIGHBOUR_TOP_RIGHT;
+                m_connections[i * m_width + (j + 1)] &= ~NEIGHBOUR_BOTTOM_LEFT;
+            }
+        }
+    }
+}