a few minor fixes in Python samples

This commit is contained in:
Vadim Pisarevsky
2011-07-13 06:19:21 +00:00
parent f098d98908
commit 386f147597
5 changed files with 17 additions and 9 deletions

View File

@@ -1,8 +1,8 @@
import numpy as np
import math
from numpy import random
import cv2
def make_gaussians(cluster_n, img_size):
points = []
ref_distrs = []
@@ -20,7 +20,7 @@ def make_gaussians(cluster_n, img_size):
def draw_gaussain(img, mean, cov, color):
x, y = np.int32(mean)
w, u, vt = cv2.SVDecomp(cov)
ang = np.rad2deg( np.arctan2(u[1, 0], u[0, 0]) )
ang = np.arctan2(u[1, 0], u[0, 0])*(180/math.pi)
s1, s2 = np.sqrt(w)*3.0
cv2.ellipse(img, (x, y), (s1, s2), ang, 0, 360, color, 1, cv2.CV_AA)