Skip to content

Instantly share code, notes, and snippets.

@stetro
Last active February 29, 2016 10:38
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save stetro/1c3d3ae032df6d2c7f0f to your computer and use it in GitHub Desktop.
Save stetro/1c3d3ae032df6d2c7f0f to your computer and use it in GitHub Desktop.
This script is used to compare the occlusion results of my master thesis prototype at https://github.com/stetro/project-tango-poc
from cv2 import *
from os import listdir
from os.path import isfile, join
reference_path = "reference.png"
reference = imread(reference_path)
reference = cvtColor(reference, COLOR_BGR2GRAY)
all_images = [f for f in listdir("./") if isfile(join("./", f)) and f.endswith(".png")]
results = []
for img_path in all_images:
img = imread(img_path)
img = cvtColor(img, COLOR_BGR2GRAY)
result = absdiff(reference, img)
imshow('res', result)
imwrite('result_'+img_path, result)
difference = sumElems(result)
results.append([reference_path, img_path, difference[0]])
waitKey(0)
for result in results:
print str(int(result[2])) + "\t" + result[0] + "\t" + result[1]

Szene 1

0       reference.png   reference.png
499793  reference.png   pc.png
441672  reference.png   pc_filtered.png
537532  reference.png   tsdf.png
0       reference.png   tsdf_filtered.png
300019  reference.png   plane.png
250254  reference.png   plane_filtered.png

Szene 2

0       reference.png   reference.png
190317  reference.png   pc.png
0       reference.png   pc_filtered.png
460392  reference.png   tsdf.png
334400  reference.png   tsdf_filtered.png
296190  reference.png   plane.png
278334  reference.png   plane_filtered.png

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment