Function:FocusStack

From CUVI Wiki

Stacks multiple images into a single, all focused image. The input images need to be aligned and in order. The function also returns a depth_info image that contains the information against each pixel where it was picked from the stack.

Function

CUVI_EXPORTS CuviStatus focusStack(CuviImage* images, const int numImages, CuviImage& stack, CuviImage& depth_info, const CuviFilter denoiseFilter, const CuviStream& stream);

Parameters

Name Type Description
images const CuviImage* Input images array
numImages const int Number of images
stack CuviImage& Resultant image
depth_info CuviImage& Slice/Depth information against stacked image
denoiseFilter CuviFilter Filter to remove noise
stream const CuviStream& GPU stream ID for execution

Image Type Support

Input Output
8uC3 x N 8uC3
16uC3 x N 16uC3

Samples

Input image with foreground focused


Input image with background focused


Focus-Stacked Image


Code Example

std::string path = "D:/dataset/fs/imageFolder/";
std::string result = "D:/dataset/fs/focused.png";


for (const auto& entry : fs::directory_iterator(path))
	{
		if (entry.path().has_extension() && isSupportedExtension(entry.path().extension().string()))
			vec.push_back(entry.path().string());
	}

int numImages = vec.size();
vector<CuviImage> ImgArr(numImages);
//Sort filesnames to read in order. THIS IS VERY IMPORTANT FOR FOCUS STACKING
sort(vec.begin(), vec.end(), lessFirst);

for (int i = 0; i < numImages; i++)
	{
		ImgArr[i].create(vec[i], CUVI_LOAD_IMAGE_COLOR_KEEP_DEPTH);
		cout << vec[i]<< std::endl;
	}


CuviImage focusImage, slice_info;
CuviFilter denoiseFilter = CuviSpecialFilters::gaussian(CuviSize(11, 11), 15.0f);
cuvi::computerVision::focusStack(ImgArr.data(), numImages, focusImage, slice_info, denoiseFilter);

printf("\nSaving to disk..");
cuvi::io::saveImage(focusImage, result);