| from PIL import Image,ImageDraw |
| from .draw_utils import box_to_xy,to_int_points,box_to_point |
| |
| def create_color_image(width, height, color=(255,255,255)): |
| if color == None: |
| color = (0,0,0) |
| |
| if len(color )== 3: |
| mode ="RGB" |
| elif len(color )== 4: |
| mode ="RGBA" |
|
|
| img = Image.new(mode, (width, height), color) |
| return img |
|
|
| |
| def fill_points(image,points,color=(255,255,255)): |
| return draw_points(image,points,fill=color) |
|
|
| def draw_points(image,points,outline=None,fill=None,width=1): |
| |
| draw = ImageDraw.Draw(image) |
| int_points = [(int(x), int(y)) for x, y in points] |
|
|
| if outline is not None or fill is not None: |
| draw.polygon(int_points, outline=outline,fill=fill,width=width) |
| |
| return image |
|
|
| def draw_box(image,box,outline=None,fill=None): |
| points = to_int_points(box_to_point(box)) |
| return draw_points(image,points,outline,fill) |
|
|
| def from_numpy(numpy_array): |
| return Image.fromarray(numpy_array) |
