bigbrush-2x2.sh

#! /bin/bash

# DEFINES
IMAGE_SIZE=1000 # -image_size for neural-style
ITERATIONS=250 # -num_iterations for neural-style
EXPAND_BORDER=50 # expanded border while cropping 

main(){
    # 1. defines
    input=$1
    input_file=`basename $input`
    clean_name="${input_file%.*}"

    style=$2
    style_dir=`dirname $style`
    style_file=`basename $style`
    style_name="${style_file%.*}"

    output="./output"
    mkdir -p $output
    out_file=$output/$input_file

    # 2. neural-style the original input
    if [ ! -s $out_file ] ; then
        neural_style $input $style $out_file
    fi
  #convert $input -resize 1000x1000 $out_file #juse resize

    # first output file size
    w1=`convert $out_file -format "%w" info:`
    h1=`convert $out_file -format "%h" info:`

    # 3. tile it
    out_dir=$output/$clean_name
    mkdir -p $out_dir
    convert $out_file -crop 2x2+$EXPAND_BORDER+$EXPAND_BORDER@ +repage +adjoin $out_dir/$clean_name"_%d.png"
    #To change the crop values, change the +20+20 on line 35, and change the "20" on lines 77,78,99,100. Make sure to use the same value.

    w2=`convert $out_dir/$clean_name'_0.png' -format "%w" info:`
    h2=`convert $out_dir/$clean_name'_0.png' -format "%h" info:`

    #Resize all tiles to avoid ImageMagick weirdness                    
    for outfile in `ls $out_dir | grep $clean_name"_"[0-9]".png"`
    do
        convert $out_dir/$outfile -resize "$w2"x"$h2"\! $out_dir/$outfile
    done

    # 4. neural-style each tile
    tiles_dir="$out_dir/tiles"
    mkdir -p $tiles_dir
    for tile in `ls $out_dir | grep $clean_name"_"[0-9]".png"`
    do
        neural_style_tiled $out_dir/$tile $style $tiles_dir/$tile
    done

    #Test   
    w2_2=`convert $tiles_dir/$clean_name'_0.png' -format "%w" info:`
    h2_2=`convert $tiles_dir/$clean_name'_0.png' -format "%h" info:`
    w_20=$EXPAND_BORDER
    h_20=$EXPAND_BORDER

    w2_22=`echo $w2 $w2_2 | awk '{print $1/$2}'`
    h2_22=`echo $h2 $h2_2 | awk '{print $1/$2}'`


    border_w_2=`echo $w2_22 $w_20 | awk '{print $1*$2}'`
    border_h_2=`echo $h2_22 $h_20 | awk '{print $1*$2}'`


    border_w=`echo $border_w_2 $w_20 | awk '{print $1+$2}'`
    border_h=`echo $border_h_2 $h_20 | awk '{print $1+$2}'`

    w_percent=`echo 20 $w2_2 | awk '{print $1/$2}'`
    h_percent=`echo 20 $h2_2 | awk '{print $1/$2}'`

    # 5. feather tiles
    feathered_dir=$out_dir/feathered
    mkdir -p $feathered_dir
    for tile in `ls $tiles_dir | grep $clean_name"_"[0-9]".png"`
    do
        tile_name="${tile%.*}"
        convert $tiles_dir/$tile -alpha set -virtual-pixel transparent -channel A -morphology Distance Euclidean:1,35\! +channel "$feathered_dir/$tile_name.png"
    done

    # 6. merge feathered tiles
    montage $feathered_dir/$clean_name'_0.png' $feathered_dir/$clean_name'_1.png' \
                    $feathered_dir/$clean_name'_2.png' $feathered_dir/$clean_name'_3.png' \
                    -tile 2x2 -geometry -$border_w-$border_h $output/$clean_name.large_feathered.png
 # -$border_w-$border_h

    # 7. merge un-feathered tiles
    montage $tiles_dir/$clean_name'_0.png' $tiles_dir/$clean_name'_1.png' \
                    $tiles_dir/$clean_name'_2.png' $tiles_dir/$clean_name'_3.png' \
                    -tile 2x2 -geometry -$border_w-$border_h $output/$clean_name.large.png

    # 8. Combine feathered and un-feathered output images to disguise feathering.
    composite $output/$clean_name.large_feathered.png $output/$clean_name.large.png $output/$clean_name.large_final.png

}

retry=0
neural_style(){
    echo "Neural Style Transfering "$1
    if [ ! -s $3 ]; then
        th neural_style.lua -content_image $1 -style_image $2 -output_image $3 \
                -image_size $IMAGE_SIZE -print_iter 100 -backend cudnn -gpu 0 -save_iter 0 \
                -style_weight 20 -num_iterations $ITERATIONS 
    fi
    if [ ! -s $3 ] && [ $retry -lt 3 ] ;then
            echo "Transfer Failed, Retrying for $retry time(s)"
            retry=`echo 1 $retry | awk '{print $1+$2}'`
            neural_style $1 $2 $3
    fi
    retry=0
}


retry=0
neural_style_tiled(){
    echo "Neural Style Transfering "$1
    if [ ! -s $3 ]; then
        th neural_style.lua -content_image $1 -style_image $2 -output_image $3 \
            -image_size $IMAGE_SIZE -print_iter 100 -backend cudnn -gpu 0 -save_iter 0 \
                -style_weight 20 -num_iterations $ITERATIONS 
    fi
    if [ ! -s $3 ] && [ $retry -lt 3 ] ;then
            echo "Transfer Failed, Retrying for $retry time(s)"
            retry=`echo 1 $retry | awk '{print $1+$2}'`
            neural_style_tiled $1 $2 $3
    fi
    retry=0
}

main $1 $2 $3