Skip to content

Latest commit

 

History

History
81 lines (58 loc) · 3.36 KB

README.md

File metadata and controls

81 lines (58 loc) · 3.36 KB
Raw

InPainting

Introduction

Ce proramme est une implémentation de la méthode InPainting. Il permet de reconstruire une image à partir d'une image masquée. Pour cela, il utilise la méthode de l'InPainting qui consiste à reconstruire l'image en utilisant les pixels autour du masque.

Installation

Pour installer le programme, il suffit de cloner le repository et d'installer les dépendances.

pip3 install matplotlib numpy progressbar2 scipy skimage python-opencv

Utilisation

L'entièreté du programme réside dans la fonction run de la classe InPainting.

Lancement par ligne de commande (Recommandé)

Pour lancer le programme, il suffit de lancer le fichier main.py avec des arguments :

python3 main.py -i image.jpg -m mask.ppm

Pour plus d'informations sur les arguments, lancer le programme avec l'argument -h.

python3 main.py -h

Help :
	-i : image path
	-m : mask path
	-p : patch size
	-r : result (save, print, return)
	-v : verbose (True, False)
	-s : save (True, False)
	-plot : plot (True, False)
	-d : distance method (SSD, SSDED, MC)
	-dis : discretisation (float)
	-t : number of thread (int)
	-g : gradient method (max_gradient, mean_gradient, closest_pixel)
	-dyn : dynamic patches (True, False)
	-h : help
Example : python main.py -i image.jpg -m mask.jpg -p 9 -r save -v True -s True -plot False -d SSDED -dis 1 -t 1 -dyn False

Lancement direct par la fonction main

Pour lancer le programme directement depuis un autre fichier python, il suffit d'importer la classe InPainting et de créer une instance de main et de lancer la fonction run.

from main import InPainting

m = Main()
main.run(<args>)

Liste des arguments

Argument Description Exemple Valeur par défaut
image_path Chemin vers l'image à reconstruire image.jpg Aucune
mask_path Chemin vers le masque de l'image mask.ppm Aucune
patch_size Longueur des côtés du patch carré (doit être impaire afin d'avoir un centre) 5 9
result Action de fin de la fonction main, "save" sauvegarde l'image dans le dossier log_image, "print" affiche l'image, "return" retourne l'image en sortie de fonction "print" "save"
verbose Affiche les informations de progression True False
save Sauvegarde les image dans log_image à chaque remplacement de patch True False
distance_method Méthode de comparaison entre deux patchs ("SSD" Somme des carrés des différences, "MC" comparaison des coulerus moyennes, "SSDED" Somme des carrés des différences auquel on ajout la distance euclidienne entre les patchs, flottant f entre 0 et 1 mix SSD*f+MC*(1-f)) "MC" "SSDED"
discretisation Distance entre chacun des patchs de comparaison, plus cette discretisation est grande, plus le programme est rapide mais moins précis 5 1
nb_thread Nombre de thread utilisé pour le calcul du meilleur patch 12 1
dynamic_patches Utilise les patchs générés en tant que patchs de remplacement True False
gradient_method Permet de définir la méthode d'évaluation du gradient pour les isophotes ("mean_gradient", "max_gradient", "closest_pixel") "mean_gradient" "max_gradient"

Image de test

Certaines images sont disponibles dans le dossier image et les masques dans le dossier mask. À une image x.jpg correspondent les masques x.ppm et x_y.ppm.