def draw_lights_on_image()

in DemoEnvMapNetMetrics.py [0:0]

• 12 lines of code
• 2 McCabe index (conditional complexity)

def draw_lights_on_image(I_,light_params):
    from matplotlib import cm
    cnt = 1
    N = len(light_params)
    I=I_.copy()
    for l_ in light_params:
        l = l_[0]
        clr = np.asarray(cm.hot(cnt)[0:3])*255
        cv2.ellipse(I, (int(l[0]), int(l[1])), (int(l[2]), int(l[3])), int(l[4]), 0, 360, clr)
        cv2.circle(I, (int(l[0]), int(l[1])), 2, clr, -1)
        cnt += 1

    return I*255