Overview
A Julia package for fast 3D image quilting simulation.
This package implements an extension to the famous Efros-Freeman algorithm for texture synthesis and transfer in computer vision. Unlike the original algorithm developed for 2D images, our method can also handle 3D masked grids and pre-existing point-data very efficiently (the fastest in the literature). For more details, please refer to our paper in Citation.
Features
Masked grids
Hard data conditioning
Soft data conditioning
Fast computation with GPUs
Installation
Get the latest stable release with Julia's package manager:
Pkg.add("ImageQuilting")
For even faster computation with GPUs, please follow the instructions in GPU support.
Usage
This package is part of the GeoStats.jl framework. Solver options are displayed below:
ImageQuilting.ImgQuilt
— Type.ImgQuilt(var₁=>param₁, var₂=>param₂, ...)
Image quilting simulation solver as described in Hoffimann et al. 2017.
Parameters
Required
TI
- Training imagetemplate
- Template size in x, y and z
Optional
overlap
- Overlap size in x, y and z (default to (1/6, 1/6, 1/6))cut
- Boundary cut algorithm (:boykov (default) or :dijkstra)path
- Simulation path (:rasterup (default), :rasterdown, :dilation, or :random)simplex
- Whether to apply or not the simplex transform (default to false)inactive
- Vector of inactive voxels (i.e. tuples (i,j,k)) in the gridsoft
- A vector of(data,dataTI)
pairstol
- Initial relaxation tolerance in (0,1] (default to 0.1)
Global parameters
Optional
threads
- Number of threads in FFT (default to number of physical CPU cores)gpu
- Whether to use the GPU or the CPU (default to false)showprogress
- Whether to show or not the estimated time duration (default to false)
Low-level API
If you are interested in using the package without GeoStats.jl, please use the following function:
ImageQuilting.iqsim
— Function.iqsim(training_image::AbstractArray,
tplsizex::Integer, tplsizey::Integer, tplsizez::Integer,
gridsizex::Integer, gridsizey::Integer, gridsizez::Integer;
overlapx::Real=1/6, overlapy::Real=1/6, overlapz::Real=1/6,
soft::AbstractVector=[], hard::HardData=HardData(), tol::Real=.1,
cut::Symbol=:boykov, path::Symbol=:rasterup, simplex::Bool=false,
nreal::Integer=1, threads::Integer=CPU_PHYSICAL_CORES,
gpu::Bool=false, debug::Bool=false, showprogress::Bool=false)
Performs image quilting simulation as described in Hoffimann et al. 2017.
Parameters
Required
training_image
can be any 3D array (add ghost dimension for 2D)tplsizex
,tplsizey
,tplsizez
is the template sizegridsizex
,gridsizey
,gridsizez
is the simulation size
Optional
overlapx
,overlapy
,overlapz
is the percentage of overlapsoft
is a vector of(data,dataTI)
pairshard
is an instance ofHardData
tol
is the initial relaxation tolerance in (0,1] (default to .1)cut
is the cut algorithm (:dijkstra
or:boykov
)path
is the simulation path (:rasterup
,:rasterdown
,:dilation
or:random
)simplex
informs whether to apply or not the simplex transform to the imagenreal
is the number of realizationsthreads
is the number of threads for the FFT (default to all CPU cores)gpu
informs whether to use the GPU or the CPUdebug
informs whether to export or not the boundary cuts and voxel reuseshowprogress
informs whether to show or not estimated time duration
The main output reals
consists of a list of 3D realizations that can be indexed with reals[1], reals[2], ..., reals[nreal]
. If debug=true
, additional output is generated:
reals, cuts, voxs = iqsim(..., debug=true)
cuts[i]
is the boundary cut for reals[i]
and voxs[i]
is the associated voxel reuse.
The major difference compared to the high-level API is that the iqsim
function has no notion of coordinate system, and you will have to pre-process the data manually to match it with the cells in the simulation grid.
GeoStats.jl takes the coordinate system into account and also enables parallel simulation on HPC clusters.