Solved visualisation bug

Author: RiSaMa

libs/_3d/projection.py

@@ -216,10 +216,11 @@ def getObjectDimensions(_c1, _c2, centroid, draw=False):
 	height = (estHeights[-1] - estHeights[0]) * 1000
 
 	# Draw final dimensions
+	img1 = copy.deepcopy(c1['rgb'])
+	img2 = copy.deepcopy(c2['rgb'])
+	
 	if draw:
-		img1 = copy.deepcopy(c1['rgb'])
-		img2 = copy.deepcopy(c2['rgb'])
-
+		
 		for i, rad in enumerate(estRadius):
 
 			p3d = []

main.py

@@ -78,7 +78,8 @@ def getObjectDimensions(self):
 			centroid = centroid[:,:-1].reshape(-1)
 
 			height, width, visualization = getObjectDimensions(self.c1, self.c2, centroid, self.args.draw)
-			cv2.imwrite('{}/id{}_l{}_b{}.png'.format(self.output_path, self.args.object, self.args.lighting, self.args.background), visualization )
+			if self.args.draw:
+				cv2.imwrite('{}/id{}_l{}_b{}.png'.format(self.output_path, self.args.object, self.args.lighting, self.args.background), visualization )
 
 			f.write('id{}_l{}_b{}.png\t{:.2f}\t{:.2f}\n'.format(self.args.object, self.args.lighting, self.args.background, height, width))
 			print('Object {}, lighting {}, and background {} measured!'.format(self.args.object, self.args.lighting, self.args.background))