| @@ -10,7 +10,7 @@ N_COLORS = 3 | |||||
| def draw_rectangle(frame, sp, ep): | def draw_rectangle(frame, sp, ep): | ||||
| """ Draw a rectangle on the frame """ | """ Draw a rectangle on the frame """ | ||||
| return cv2.rectangle(frame, sp, ep, RED) | |||||
| return cv2.rectangle(frame, sp, ep, RED, 1) | |||||
| def analyze_block(frame, index, sp, ep): | def analyze_block(frame, index, sp, ep): | ||||
| @@ -20,10 +20,18 @@ def analyze_block(frame, index, sp, ep): | |||||
| sp2 = tuple(int(x) for x in (2 * np.array(sp) + np.array(ep)) / 3) | sp2 = tuple(int(x) for x in (2 * np.array(sp) + np.array(ep)) / 3) | ||||
| ep2 = tuple(int(x) for x in (np.array(sp) + 2 * np.array(ep)) / 3) | ep2 = tuple(int(x) for x in (np.array(sp) + 2 * np.array(ep)) / 3) | ||||
| average_color = [int(x) for x in average_color] | average_color = [int(x) for x in average_color] | ||||
| frame = cv2.rectangle(frame, sp2, ep2, average_color, 5) | |||||
| frame = cv2.rectangle(frame, sp2, ep2, average_color, -1) | |||||
| average_color = np.uint8([[average_color]]) | average_color = np.uint8([[average_color]]) | ||||
| h, s, v = cv2.cvtColor(average_color, cv2.COLOR_BGR2HSV)[0][0] | h, s, v = cv2.cvtColor(average_color, cv2.COLOR_BGR2HSV)[0][0] | ||||
| print(index, h, s, v) | |||||
| for thickness, color in ((3, (0, 0, 0)), (1, (255, 255, 255))): | |||||
| frame = cv2.putText(frame, | |||||
| text=f"{h:03d} {s:03d} {v:03d}", | |||||
| org=(sp[0] + 10, sp[1] + 20), | |||||
| fontScale=.5, | |||||
| color=color, | |||||
| fontFace=2, | |||||
| thickness=thickness) | |||||
| return frame | return frame | ||||