Commit 46dd85a

authored

Fix drawBoxesOnRGB with RGB images

fix #22

1 parent a0de0e5 commit 46dd85aCopy full SHA for 46dd85a

File tree

-2

lines changed

-2

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -324,7 +324,7 @@ def drawBoxesOnRGB(image:NDArray, listHit:Sequence[Hit], boxThickness:int=2, box`
`324`	`324`	`original image with predicted template locations depicted as bounding boxes`
`325`	`325`	`"""`
`326`	`326`	`# Convert Grayscale to RGB to be able to see the color bboxes`
`327`		`- outImage = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB) if image.ndim == 2 else image.copy`
	`327`	`+ outImage = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB) if image.ndim == 2 else image.copy()`
`328`	`328`
`329`	`329`	`for label, bbox, _ in listHit:`
`330`	`330`
`@@ -388,4 +388,4 @@ def drawBoxesOnGray(image:NDArray, listHit:Sequence[Hit], boxThickness=2, boxCol`
`388`	`388`	`color=labelColor,`
`389`	`389`	`lineType=cv2.LINE_AA)`
`390`	`390`
`391`		`- return outImage`
	`391`	`+ return outImage`

Comments

(0)