m1-ENSL/seam-carving
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adds remove_seam function

Browse Source
This commit is contained in:
François Colin de Verdière 2025-04-01 11:51:56 +02:00
parent 964d597bdd
commit f15b4c471b
1 changed files with 184 additions and 127 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

305
src/seam-carving.cpp
View File

@ -1,103 +1,113 @@
#include <iostream> #include <iostream>
#include <string>
#include <random> #include <random>
#include <string>
#include <vector> #include <vector>
//Command-line parsing // Command-line parsing
#include <CLI11.hpp> #include <CLI11.hpp>
//Image filtering and I/O // Image filtering and I/O
#define STB_IMAGE_IMPLEMENTATION #define STB_IMAGE_IMPLEMENTATION
#include <stb_image.h> #include <stb_image.h>
#define STB_IMAGE_WRITE_IMPLEMENTATION #define STB_IMAGE_WRITE_IMPLEMENTATION
#include <stb_image_write.h>
#include <SimpleProgressBar.hpp> #include <SimpleProgressBar.hpp>
#include <stb_image_write.h>
//Global flag to silent verbose messages // Global flag to silent verbose messages
bool silent; bool silent;
bool test_energy; bool test_energy;
#define min(a, b) { (a < b ? a : b) } #define min(a, b) \
{ (a < b ? a : b) }
// Get index for any table indexed by [width*(i : height) + (j : width)], but a : dim_long, b : dim_large
#define im_index(a, b) \
(vertical ? (width*a + b) : (width*b + a))
// Get index for any table indexed by [width*(i : height) + (j : width)], but a
// : dim_long, b : dim_large
#define im_index(a, b) (vertical ? (width * a + b) : (width * b + a))
bool nearly_equal(float a, float b) { bool nearly_equal(float a, float b) {
return std::nextafter(a, std::numeric_limits<float>::lowest()) <= b return std::nextafter(a, std::numeric_limits<float>::lowest()) <= b &&
&& std::nextafter(a, std::numeric_limits<float>::max()) >= b; std::nextafter(a, std::numeric_limits<float>::max()) >= b;
} }
void export_image(const char* filename, const void* data, int width, int height, int nbChannels) { void export_image(const char *filename, const void *data, int width, int height,
if (!silent) std::cout << "Exporting to \"" << filename << "\".." << std::endl; int nbChannels) {
int errcode = stbi_write_png(filename, width, height, nbChannels, data, nbChannels*width); if (!silent)
std::cout << "Exporting to \"" << filename << "\".." << std::endl;
int errcode = stbi_write_png(filename, width, height, nbChannels, data,
nbChannels * width);
if (!errcode) { if (!errcode) {
std::cerr << "Error while exporting the resulting image." << std::endl; std::cerr << "Error while exporting the resulting image." << std::endl;
exit(errcode); exit(errcode);
} }
} }
/** e_1 energy, energy is always normalized between 0 and 1 */ /** e_1 energy, energy is always normalized between 0 and 1 */
std::vector<float> energy_e1(std::vector<unsigned char> source, int width, int height, int nbChannels) { std::vector<float> energy_e1(std::vector<unsigned char> source, int width,
int nbColorChannels = nbChannels > 3 ? 3 : nbChannels; // nombre de canaux, excepté le alpha int height, int nbChannels) {
std::vector<float> energy(width*height); int nbColorChannels =
nbChannels > 3 ? 3 : nbChannels; // nombre de canaux, excepté le alpha
std::vector<float> energy(width * height);
float max_energy = 0; float max_energy = 0;
for(auto i=0 ; i < width; ++i) { for (auto i = 0; i < width; ++i) {
for(auto j=0; j < height; ++j) { for (auto j = 0; j < height; ++j) {
auto indexPixel = nbChannels*(width*j+i); auto indexPixel = nbChannels * (width * j + i);
auto indexPixel_up = (j-1 > 0) ? nbChannels*(width*(j-1)+i) : indexPixel; auto indexPixel_up =
auto indexPixel_down = (j+1 < height) ? nbChannels*(width*(j+1)+i) : indexPixel; (j - 1 > 0) ? nbChannels * (width * (j - 1) + i) : indexPixel;
auto indexPixel_left = (i-1 > 0) ? nbChannels*(width*j+(i-1)) : indexPixel; auto indexPixel_down =
auto indexPixel_right = (i+1 < width) ? nbChannels*(width*j+(i+1)) : indexPixel; (j + 1 < height) ? nbChannels * (width * (j + 1) + i) : indexPixel;
auto indexPixel_left =
energy[width*j+i] = 0; (i - 1 > 0) ? nbChannels * (width * j + (i - 1)) : indexPixel;
for (auto ch=0; ch < nbColorChannels; ch++) { // Le alpha n'est pas pris en compte dans l'énergie auto indexPixel_right =
energy[width*j+i] += ( (i + 1 < width) ? nbChannels * (width * j + (i + 1)) : indexPixel;
fabs((float)source[indexPixel_up+ch] - source[indexPixel+ch]) energy[width * j + i] = 0;
+fabs((float)source[indexPixel_down+ch] - source[indexPixel+ch]) for (auto ch = 0; ch < nbColorChannels;
+fabs((float)source[indexPixel_left+ch] - source[indexPixel+ch]) ch++) { // Le alpha n'est pas pris en compte dans l'énergie
+fabs((float)source[indexPixel_right+ch] - source[indexPixel+ch]) energy[width * j + i] +=
); (fabs((float)source[indexPixel_up + ch] - source[indexPixel + ch]) +
fabs((float)source[indexPixel_down + ch] -
source[indexPixel + ch]) +
fabs((float)source[indexPixel_left + ch] -
source[indexPixel + ch]) +
fabs((float)source[indexPixel_right + ch] -
source[indexPixel + ch]));
} }
max_energy = fmax(max_energy, energy[width*j+i]); max_energy = fmax(max_energy, energy[width * j + i]);
} }
} }
if (max_energy == 0) { return energy; } if (max_energy == 0) {
for (auto k=0; k < width*height; k++) { return energy;
energy[k] = energy[k]/max_energy; }
for (auto k = 0; k < width * height; k++) {
energy[k] = energy[k] / max_energy;
} }
return energy; return energy;
} }
/** Given the energy value, returns the optimal seam */ std::vector<int> optimal_seam(std::vector<float> energy, int width, int height,
std::vector<int> optimal_seam(std::vector<float> energy, int width, int height, bool vertical) { bool vertical) {
/** Given the energy value, returns the optimal seam */
// dyn_energy is indexed by [dim_large*(i : dim_long) + (j : dim_large)] // dyn_energy is indexed by [dim_large*(i : dim_long) + (j : dim_large)]
std::vector<float> dyn_energy(width*height); std::vector<float> dyn_energy(width * height);
int dim_large = vertical ? width : height; int dim_large = vertical ? width : height;
int dim_long = vertical ? height : width; // Number of elements in the seam int dim_long = vertical ? height : width; // Number of elements in the seam
//* Find an end of the minimal connected vertical/horizontal seam //* Find an end of the minimal connected vertical/horizontal seam
for (auto i=0; i < dim_large; i++) { for (auto i = 0; i < dim_large; i++) {
dyn_energy[i] = energy[i]; dyn_energy[i] = energy[i];
} }
for (auto i=1; i < dim_long; i++) { // Propagate dyn_energy for (auto i = 1; i < dim_long; i++) { // Propagate dyn_energy
for (auto j=0; j < dim_large; j++) { for (auto j = 0; j < dim_large; j++) {
float bot_center = dyn_energy[dim_large*(i-1) + j]; float bot_center = dyn_energy[dim_large * (i - 1) + j];
float bot_left = (j > 0) ? dyn_energy[dim_large*(i-1) + (j-1)] : __FLT_MAX__; float bot_left =
float bot_right = (j+1 < dim_large) ? dyn_energy[dim_large*(i-1) + (j+1)] : __FLT_MAX__; (j > 0) ? dyn_energy[dim_large * (i - 1) + (j - 1)] : __FLT_MAX__;
float bot_right = (j + 1 < dim_large)
? dyn_energy[dim_large * (i - 1) + (j + 1)]
: __FLT_MAX__;
dyn_energy[dim_large*i+j] = fmin( dyn_energy[dim_large * i + j] =
bot_center, fmin(bot_center, fmin(bot_left, bot_right)) + energy[im_index(i, j)];
fmin(
bot_left,
bot_right
)
) + energy[im_index(i, j)];
} }
} }
@ -105,111 +115,146 @@ std::vector<int> optimal_seam(std::vector<float> energy, int width, int height,
// Find the seam end // Find the seam end
int min_idx = -1; int min_idx = -1;
float min_val = __FLT_MAX__; float min_val = __FLT_MAX__;
for (auto j=0; j < dim_large; j++) { for (auto j = 0; j < dim_large; j++) {
if (min_val > dyn_energy[dim_large*(dim_long-1) + j]) { if (min_val > dyn_energy[dim_large * (dim_long - 1) + j]) {
min_idx = j; min_idx = j;
min_val = dyn_energy[dim_large*(dim_long-1) + j]; min_val = dyn_energy[dim_large * (dim_long - 1) + j];
} }
} }
result[dim_long-1] = min_idx; result[dim_long - 1] = min_idx;
//* Backtracking to find the path //* Backtracking to find the path
for (auto i=dim_long-1; i > 0; i--) { for (auto i = dim_long - 1; i > 0; i--) {
// We want to find either (bot_l, bot_c, bot_r) with dyn_energy[.] = min_val - energy[cur] // We want to find either (bot_l, bot_c, bot_r) with dyn_energy[.] = min_val -
// energy[cur]
//Idea : float next_energy = min_val - energy[width*i + min_idx]; // Idea : float next_energy = min_val - energy[width*i + min_idx];
//! With floats, we don't always have x + y - y == x, so we check is x+y == x+y //! With floats, we don't always have x + y - y == x, so we check is x+y == x+y
// This define is a bit ugly but 200x faster than using a lambda function // This define is a bit ugly but 200x faster than using a lambda function
#define is_next_idx(idx) \ #define is_next_idx(idx) \
(dyn_energy[(i-1)*dim_large + idx]+energy[im_index(i, min_idx)] == min_val) (dyn_energy[(i - 1) * dim_large + idx] + energy[im_index(i, min_idx)] == \
min_val)
if (is_next_idx(min_idx)) { if (is_next_idx(min_idx)) {
// min_idx does not change // min_idx does not change
min_val = dyn_energy[(i-1)*dim_large + min_idx]; min_val = dyn_energy[(i - 1) * dim_large + min_idx];
} else if (min_idx > 0 && is_next_idx(min_idx-1)) { } else if (min_idx > 0 && is_next_idx(min_idx - 1)) {
min_val = dyn_energy[(i-1)*dim_large + (min_idx-1)]; min_val = dyn_energy[(i - 1) * dim_large + (min_idx - 1)];
min_idx = min_idx - 1; min_idx = min_idx - 1;
} else if (min_idx+1 < dim_large && is_next_idx(min_idx+1)) { } else if (min_idx + 1 < dim_large && is_next_idx(min_idx + 1)) {
min_val = dyn_energy[(i-1)*dim_large + (min_idx+1)]; min_val = dyn_energy[(i - 1) * dim_large + (min_idx + 1)];
min_idx = min_idx + 1; min_idx = min_idx + 1;
} else { } else {
std::cerr << "Unable to backtrack path !" << std::endl; std::cerr << "Unable to backtrack path !" << std::endl;
exit(1); exit(1);
} }
result[i-1] = min_idx; result[i - 1] = min_idx;
} }
return result; return result;
} }
/** Carves an image by one seam. Returns the optimal seam used */
std::vector<int> carving_step(const std::vector<unsigned char> source, std::vector<unsigned char> &output, int width, int height, int nbChannels, bool vertical) { void remove_seam(const std::vector<unsigned char> source,
std::vector<float> energy = energy_e1(source, width, height, nbChannels); std::vector<unsigned char> &output, int width, int height,
std::vector<int> opt_seam = optimal_seam(energy, width, height, vertical); int nbChannels, bool vertical, const std::vector<int> seam)
std::vector<bool> blacklist(width*height); {
/* remove the given seam from the image, the result is in output*/
std::vector<bool> blacklist(width * height);
int dim_large = vertical ? width : height; int dim_large = vertical ? width : height;
int dim_long = vertical ? height : width; int dim_long = vertical ? height : width;
for (auto k=0; k < width*height; k++) { blacklist[k] = false; } for (auto k = 0; k < width * height; k++) {
for (auto i=0; i < dim_long; i++) { blacklist[k] = false;
int index = vertical ? opt_seam[i]+i*width : i+width*opt_seam[i]; }
for (auto i = 0; i < dim_long; i++) {
int index = vertical ? seam[i] + i * width : i + width * seam[i];
blacklist[index] = true; blacklist[index] = true;
} }
for (auto i=0; i < dim_long; i++) { for (auto i = 0; i < dim_long; i++) {
int cur_j = 0; int cur_j = 0;
for (auto j=0; cur_j < dim_large-1 && j < dim_large; j++) { for (auto j = 0; cur_j < dim_large - 1 && j < dim_large; j++) {
if (!blacklist[im_index(i, j)]) { if (!blacklist[im_index(i, j)]) {
int out_pixelIndex = nbChannels*(vertical ? ((width-1)*i + cur_j) : (width*cur_j + i)); int out_pixelIndex = nbChannels * (vertical ? ((width - 1) * i + cur_j)
int src_pixelIndex = nbChannels*im_index(i, j); : (width * cur_j + i));
int src_pixelIndex = nbChannels * im_index(i, j);
for (auto ch=0; ch < nbChannels; ch++) for (auto ch = 0; ch < nbChannels; ch++)
output[out_pixelIndex+ch] = source[src_pixelIndex+ch]; output[out_pixelIndex + ch] = source[src_pixelIndex + ch];
cur_j++; cur_j++;
} }
} }
} }
}
std::vector<int> carving_step(const std::vector<unsigned char> source,
std::vector<unsigned char> &output, int width,
int height, int nbChannels, bool vertical) {
/** Carves an image by one seam. Returns the optimal seam used */
std::vector<float> energy = energy_e1(source, width, height, nbChannels);
std::vector<int> opt_seam = optimal_seam(energy, width, height, vertical);
remove_seam(source, output, width, height, nbChannels, vertical, opt_seam);
return opt_seam; return opt_seam;
} }
void seam_carving(unsigned char* source, int width, int height, int nbChannels, const char* out_filename, int nbSeams, bool vertical, bool test_energy=false) {
void seam_carving(unsigned char *source, int width, int height, int nbChannels,
const char *out_filename, int nbSeams, bool vertical,
bool test_energy = false) {
int nbColorChannels = nbChannels > 3 ? 3 : nbChannels; int nbColorChannels = nbChannels > 3 ? 3 : nbChannels;
int curWidth = width; int curWidth = width;
int curHeight = height; int curHeight = height;
int dim_large = vertical ? width : height; int dim_large = vertical ? width : height;
int dim_long = vertical ? height : width; int dim_long = vertical ? height : width;
std::vector<unsigned char> carve_output(width*height*nbChannels); // Receives at each step the newly carved image std::vector<unsigned char> carve_output(
std::vector<unsigned char> source_img(width*height*nbChannels); // Contains at each step the carved image width * height *
std::vector<bool> complete_blacklist(width*height); // Contains all removed pixels, for "test_energy" nbChannels); // Receives at each step the newly carved image
std::vector<float> ini_energy; // Contains the initial energy, only for "test_energy" std::vector<unsigned char> source_img(
std::vector<unsigned char> test_energy_output(width*height*nbChannels); // Final output for "test_energy" width * height * nbChannels); // Contains at each step the carved image
std::vector<bool> complete_blacklist(
width * height); // Contains all removed pixels, for "test_energy"
std::vector<float>
ini_energy; // Contains the initial energy, only for "test_energy"
std::vector<unsigned char> test_energy_output(
width * height * nbChannels); // Final output for "test_energy"
for (auto i = 0; i < width * height * nbChannels; i++) {
for (auto i=0; i < width*height*nbChannels; i++) { source_img[i] = source[i]; } source_img[i] = source[i];
}
if (test_energy) { if (test_energy) {
ini_energy = energy_e1(source_img, width, height, nbChannels); ini_energy = energy_e1(source_img, width, height, nbChannels);
for (auto k=0; k < width*height; k++) { complete_blacklist[k] = false; } for (auto k = 0; k < width * height; k++) {
complete_blacklist[k] = false;
}
//* Prepare final output //* Prepare final output
for (auto k=0; k < width*height; k++) { for (auto k = 0; k < width * height; k++) {
//for (auto i=0; i < nbColorChannels; i++) //* Uncomment if you prefer to see darkened source image // for (auto i=0; i < nbColorChannels; i++) //* Uncomment if you prefer to
// output[nbChannels*k+i] = source_img[nbChannels*k+i]/nbChannels; // see darkened source image
for (auto i=0; i < nbColorChannels; i++) // output[nbChannels*k+i] = source_img[nbChannels*k+i]/nbChannels;
test_energy_output[nbChannels*k+i] = ini_energy[k]*255; for (auto i = 0; i < nbColorChannels; i++)
test_energy_output[nbChannels * k + i] = ini_energy[k] * 255;
if (nbChannels==4) if (nbChannels == 4)
test_energy_output[nbChannels*k+3] = source_img[nbChannels*k+3]; test_energy_output[nbChannels * k + 3] = source_img[nbChannels * k + 3];
} }
} }
SimpleProgressBar::ProgressBar bar(nbSeams); SimpleProgressBar::ProgressBar bar(nbSeams);
for (auto seam=0; seam < nbSeams; seam++) { for (auto seam = 0; seam < nbSeams; seam++) {
std::vector<int> opt_seam = carving_step(source_img, carve_output, curWidth, curHeight, nbChannels, vertical); std::vector<int> opt_seam = carving_step(source_img, carve_output, curWidth,
curHeight, nbChannels, vertical);
std::copy(carve_output.begin(), carve_output.end(), source_img.begin()); std::copy(carve_output.begin(), carve_output.end(), source_img.begin());
if (vertical) // We just reduced the dimension if (vertical) // We just reduced the dimension
@ -218,15 +263,20 @@ void seam_carving(unsigned char* source, int width, int height, int nbChannels,
curHeight--; curHeight--;
if (test_energy) { // Update blacklist if (test_energy) { // Update blacklist
for (auto i=0; i < dim_long; i++) { for (auto i = 0; i < dim_long; i++) {
int j, cur_j = 0; // cur_j is the index relative to the current carved image. j is absolute in the source image int j, cur_j = 0; // cur_j is the index relative to the current carved
for (j=0; j < dim_large && cur_j < opt_seam[i]; j++) { // image. j is absolute in the source image
if (!complete_blacklist[im_index(i, j)]) { cur_j++; } for (j = 0; j < dim_large && cur_j < opt_seam[i]; j++) {
if (!complete_blacklist[im_index(i, j)]) {
cur_j++;
}
} }
assert(cur_j == opt_seam[i]); // Else, j == width and cur_j is not in the source image.. assert(cur_j == opt_seam[i]); // Else, j == width and cur_j is not in
// the source image..
complete_blacklist[im_index(i, j)] = true; complete_blacklist[im_index(i, j)] = true;
test_energy_output[nbChannels*im_index(i, j)] = 255; // Set carved pixel to red test_energy_output[nbChannels * im_index(i, j)] =
255; // Set carved pixel to red
} }
} }
bar.increment(); bar.increment();
@ -235,18 +285,22 @@ void seam_carving(unsigned char* source, int width, int height, int nbChannels,
std::cout << std::endl; std::cout << std::endl;
if (test_energy) { if (test_energy) {
export_image(out_filename, test_energy_output.data(), width, height, nbChannels); export_image(out_filename, test_energy_output.data(), width, height,
nbChannels);
} else { } else {
export_image(out_filename, source_img.data(), curWidth, curHeight, nbChannels); export_image(out_filename, source_img.data(), curWidth, curHeight,
nbChannels);
} }
} }
int main(int argc, char **argv) { int main(int argc, char **argv) {
CLI::App app{"seam-carving"}; CLI::App app{"seam-carving"};
std::string sourceImage; std::string sourceImage;
app.add_option("-s,--source", sourceImage, "Source image")->required()->check(CLI::ExistingFile);; app.add_option("-s,--source", sourceImage, "Source image")
std::string outputImage= "output.png"; ->required()
->check(CLI::ExistingFile);
;
std::string outputImage = "output.png";
app.add_option("-o,--output", outputImage, "Output image")->required(); app.add_option("-o,--output", outputImage, "Output image")->required();
int nbSeams = 1; int nbSeams = 1;
app.add_option("-n,--nb-seams", nbSeams, "Number of seams"); app.add_option("-n,--nb-seams", nbSeams, "Number of seams");
@ -255,17 +309,20 @@ int main(int argc, char **argv) {
silent = false; silent = false;
app.add_flag("--silent", silent, "No verbose messages"); app.add_flag("--silent", silent, "No verbose messages");
test_energy = false; test_energy = false;
app.add_flag("--test-energy", test_energy, "Don't resize image, just try the specified energy function"); app.add_flag("--test-energy", test_energy,
"Don't resize image, just try the specified energy function");
CLI11_PARSE(app, argc, argv); CLI11_PARSE(app, argc, argv);
//Image loading // Image loading
int width, height, nbChannels; int width, height, nbChannels;
unsigned char *source = stbi_load(sourceImage.c_str(), &width, &height, &nbChannels, 0); unsigned char *source =
stbi_load(sourceImage.c_str(), &width, &height, &nbChannels, 0);
nbSeams = min(nbSeams, width); nbSeams = min(nbSeams, width);
//std::cout << "channels: " << nbChannels << std::endl; // std::cout << "channels: " << nbChannels << std::endl;
seam_carving(source, width, height, nbChannels, outputImage.c_str(), nbSeams, vertical, test_energy=test_energy); seam_carving(source, width, height, nbChannels, outputImage.c_str(), nbSeams,
vertical, test_energy = test_energy);
stbi_image_free(source); stbi_image_free(source);
exit(0); exit(0);