Commit da33b15

committed

Use square bounding box

1 parent f84dd0b commit da33b15Copy full SHA for da33b15

File tree

-7

lines changed

-7

lines changed

Lines changed: 8 additions & 7 deletions

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@`
`15`	`15`	`OPENPOSE_ROOT = os.environ["OPENPOSE_ROOT"]`
`16`	`16`
`17`	`17`
`18`		`-def ComputeBB(hand, padding=0.4):`
	`18`	`+def ComputeBB(hand, padding=1.5):`
`19`	`19`	`minX = np.min(hand[:, 0])`
`20`	`20`	`minY = np.min(hand[:, 1])`
`21`	`21`
`@@ -25,14 +25,15 @@ def ComputeBB(hand, padding=0.4):`
`25`	`25`	`width = maxX - minX`
`26`	`26`	`height = maxY - minY`
`27`	`27`
`28`		`- padX = width*padding/2`
`29`		`- padY = height*padding/2`
	`28`	`+ cx = minX+width/2`
	`29`	`+ cy = minY+height/2`
`30`	`30`
`31`		`- minX -= padX`
`32`		`- minY -= padY`
	`31`	`+ width = height = max(width, height)`
	`32`	`+ width = height = width * padding`
	`33`	`+`
	`34`	`+ minX = cx - width/2`
	`35`	`+ minY = cy - height/2`
`33`	`36`
`34`		`- width += 2 * padX`
`35`		`- height += 2 * padY`
`36`	`37`
`37`	`38`	`score = np.mean(hand[:, 2])`
`38`	`39`	`return score, [int(minX), int(minY), int(width), int(height)]`

Comments

(0)