denoising.py/code/main.py

219 lines
7.2 KiB
Python
Raw Normal View History

2015-09-06 12:54:58 +00:00
# ~*~ encoding: utf-8 ~*~ #
###########################
# TRAITEMENT D'IMAGES #
###########################
2015-09-06 14:01:17 +00:00
# classes
2015-09-09 13:52:20 +00:00
from BMPFile import *
2015-09-10 21:37:17 +00:00
from Noise import *
from tests import *
import random
2015-09-06 12:54:58 +00:00
import sys
import time
2015-10-02 12:10:13 +00:00
# test par défaut puis quitte
# defaultTest();
# exit();
2015-09-17 22:33:14 +00:00
# arrêt si moins de 2 arguments
if len(sys.argv) < 3:
print "Require 2 args : \n* input image\n* output image"
exit()
2015-10-02 12:10:13 +00:00
2015-09-17 22:33:14 +00:00
################" INTERFACE "###################
2015-09-18 11:55:22 +00:00
print "+---------------------------+"
print "| |"
print "| TRAITEMENT D'IMAGE |"
print "| |"
print "+---------------------------+"
print "| <in> %s |" % exactLength( sys.argv[1], 19, -1)
print "| <out> %s |" % exactLength( sys.argv[2], 19, -1)
print "+---------------------------+"
print "| %s |" % exactLength("TESTS DE FICHIER", 25, 0)
print "| %s |" % exactLength("", 25, 0)
print "| 0) %s |" % exactLength("Creation manuelle", 21, -1)
print "| 1) %s |" % exactLength("Parse/Unparse", 21, -1)
print "| 2) %s |" % exactLength("Afficher palette", 21, -1)
print "+---------------------------+"
print "| %s |" % exactLength("TESTS DE BRUIT", 25, 0)
print "| %s |" % exactLength("", 25, 0)
print "| 10) %s |" % exactLength("Salt&Pepper", 21, -1)
print "| 11) %s |" % exactLength("Additif (Bernouilli)", 21, -1)
print "| 12) %s |" % exactLength("Additif (Gaussien)", 21, -1)
2015-09-18 11:55:22 +00:00
print "+---------------------------+"
print "| %s |" % exactLength("TESTS DE DIFFERENCES", 25, 0)
print "| %s |" % exactLength("", 25, 0)
print "| 20) %s |" % exactLength("Difference en %", 21, -1)
2015-09-28 14:54:23 +00:00
print "| 21) %s |" % exactLength("SNR (origine/bruit)", 21, -1)
2015-09-18 11:55:22 +00:00
print "| 22) %s |" % exactLength("Difference en image", 21, -1)
print "| 23) %s |" % exactLength("Fusion d'images (+)", 21, -1)
print "| 24) %s |" % exactLength("Fusion d'images (-)", 21, -1)
print "+---------------------------+"
print "| %s |" % exactLength("TESTS DE FORMES", 25, 0)
print "| %s |" % exactLength("", 25, 0)
print "| 30) %s |" % exactLength("Reveler une teinte", 21, -1)
print "| 31) %s |" % exactLength("Colorer une forme", 21, -1)
print "| 32) %s |" % exactLength("Colorer les formes", 21, -1)
2015-09-24 17:52:49 +00:00
print "| 33) %s |" % exactLength("Relever les contours", 21, -1)
print "+---------------------------+"
print "| %s |" % exactLength("TESTS DE FILTRES", 25, 0)
print "| %s |" % exactLength("", 25, 0)
2015-10-02 12:10:13 +00:00
print "| 40) %s |" % exactLength("Filtre moyen", 21, -1)
print "| 41) %s |" % exactLength("Laplace", 21, -1)
print "| 42) %s |" % exactLength("Roberts", 21, -1)
print "| 43) %s |" % exactLength("Prewitt", 21, -1)
print "| 44) %s |" % exactLength("Sobel", 21, -1)
print "| 45) %s |" % exactLength("Convolution", 21, -1)
print "| 46) %s |" % exactLength("bichrome", 21, -1)
print "+---------------------------+"
2015-09-17 22:33:14 +00:00
print
while True:
action = int( raw_input("choix: ") )
if action >= 0 and action < 50:
2015-09-17 22:33:14 +00:00
break;
startStr = "\n+---------------------------+---------+"
result = ""
2015-09-17 22:33:14 +00:00
execTime = Timer(); execTime.reset();
2015-09-18 11:55:22 +00:00
# fichier
2015-09-17 22:33:14 +00:00
if action == 0:
w = raw_input("width [100]: ")
h = raw_input("height [100]: ")
arg1, arg2 = 100, 100
if w != "":
arg1 = int(w)
if h != "":
arg2 = int(h)
print startStr
2015-10-02 12:10:13 +00:00
testManualCreation(arg1, arg2) # teste la création d'un fichier à partir d'une matrice uniquement
2015-09-17 22:33:14 +00:00
elif action == 1:
print startStr
2015-10-02 12:10:13 +00:00
result = testFileIntegrity() # teste le PARSE/UNPARSE
2015-09-17 22:33:14 +00:00
elif action == 2:
print startStr
2015-10-02 12:10:13 +00:00
result = printIntPalette() # affiche la palette d'une image
2015-09-18 11:55:22 +00:00
# bruits
elif action == 10:
inS = raw_input("seuil bruitage [50]: ")
outS = raw_input("seuil débruitage [1] : ")
outB = raw_input("borne débruitage [1] : ")
s = raw_input("Lissage ? (0-1): ")
arg1, arg2, arg3, arg4 = 50, 1, 1, 1
if inS != "":
arg1 = int(inS)
if outS != "":
arg2 = int(outS)
if outB != "":
arg3 = int(outB)
if s != "":
arg4 = int(s)
print startStr
2015-10-02 12:10:13 +00:00
testSaltAndPepper(arg1, arg2, arg3, arg4) # teste le bruitage/débruitage de type "Sel & Poivre"
2015-09-18 11:55:22 +00:00
elif action == 11:
inS = raw_input("seuil bruitage [10]: ")
outS = raw_input("seuil débruitage [35] : ")
arg1, arg2 = 10, 35
if inS != "":
arg1 = int(inS)
if outS != "":
arg2 = int(outS)
print startStr
2015-10-02 12:10:13 +00:00
testAdditiveBernouilliNoise(arg1, arg2) # teste le bruitage/débruitage de type "Additif"
elif action == 12:
inS = raw_input("sigma [10]: ")
outS = raw_input("seuil débruitage [35] : ")
arg1, arg2 = 10, 35
if inS != "":
arg1 = int(inS)
if outS != "":
arg2 = int(outS)
print startStr
2015-10-02 12:10:13 +00:00
testAdditiveGaussianNoise(arg1, arg2) # teste le bruitage/débruitage de type "Additif"
2015-09-18 11:55:22 +00:00
# performances
elif action == 20:
print startStr
2015-10-02 12:10:13 +00:00
printImageQuality() # compare 2 images et donne le pourcentage de ressemblance/différence
2015-09-18 11:55:22 +00:00
elif action == 21:
print startStr
2015-10-02 12:10:13 +00:00
printSNR() # compare 2 images et retourne le SNR
2015-09-18 11:55:22 +00:00
elif action == 22:
print startStr
2015-10-02 12:10:13 +00:00
imageForImageQuality() # crée une image correspondant aux différences de 2 images
2015-09-18 11:55:22 +00:00
elif action == 23:
print startStr
2015-10-02 12:10:13 +00:00
mergeImagesAdditive() # crée une image étant la fusion (addition) de 2 images
2015-09-18 11:55:22 +00:00
elif action == 24:
print startStr
2015-10-02 12:10:13 +00:00
mergeImagesSubstractive() # crée une image étant la fusion (soustractive) de 2 images
elif action == 30:
r = raw_input("rouge [0]: ")
g = raw_input("vert [0]: ")
b = raw_input("bleu [0]: ")
s = raw_input("seuil [50]: ")
arg1, arg2, arg3, arg4 = 0,0,0,50
if r != "":
arg1 = int(r)
if g != "":
arg2 = int(g)
if b != "":
arg3 = int(b)
if s != "":
arg4 = int(s)
print startStr
2015-10-02 12:10:13 +00:00
revealShapes(arg1, arg2, arg3, arg4) # révèle la couleur spécifiée
elif action == 31:
x = raw_input("abscisses(x) [0]: ")
y = raw_input("ordonnées(y) [0]: ")
arg1, arg2 = 0,0
if x != "":
arg1 = int(x)
if y != "":
arg2 = int(y)
print startStr
2015-10-02 12:10:13 +00:00
colorShape(arg1, arg2) # colorie la forme contenant le pixel de coordonnées donné
elif action == 32:
print startStr
2015-10-02 12:10:13 +00:00
colorAllShapes() # colorie la forme contenant le pixel de coordonnées donné
2015-09-24 17:52:49 +00:00
elif action == 33:
print startStr
2015-10-02 12:10:13 +00:00
testStroke() # trace les contours uniquement à partir de formes pleines
# filtres
elif action == 40:
print startStr
2015-10-02 12:10:13 +00:00
testAverageFilter() # teste le lissage
elif action == 41:
print startStr
2015-10-02 12:10:13 +00:00
testLaplace() # teste le filtre de Laplace
elif action == 42:
print startStr
2015-10-02 12:10:13 +00:00
testRoberts() # teste le filtre de Roberts
elif action == 43:
print startStr
2015-10-02 12:10:13 +00:00
testPrewitt() # teste le filtre de Prewitt
elif action == 44:
print startStr
2015-10-02 12:10:13 +00:00
testSobel() # teste le filtre de Sobel
elif action == 45:
print startStr
2015-10-02 12:10:13 +00:00
testConvolution() # teste le filtre de Convolution
elif action == 46:
print startStr
2015-10-02 12:10:13 +00:00
testBichrome() # teste le passage au bichromatique
2015-09-17 22:33:14 +00:00
else:
2015-09-18 11:55:22 +00:00
print "Wrong choice"
exit();
2015-09-17 22:33:14 +00:00
2015-09-18 11:55:22 +00:00
print "+---------------------------+---------+"
print "| EXECUTION TIME | %s |" % execTime.get()
print "+---------------------------+---------+"
print
2015-10-02 12:10:13 +00:00
print result