102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
import glob
|
|
|
|
import numpy
|
|
from PIL import Image, ImageDraw
|
|
|
|
|
|
def extract_image(filename: str, output_path: str) -> None:
|
|
"""
|
|
Extract a field card from a screenshot from a device. Screenshot for each card
|
|
taken from the Hyakabun scrollery. Image will be resized and the card extracted.
|
|
|
|
Args:
|
|
filename (str): Full path to the input image.
|
|
output_path (str): Full path to the directory where the image should be saved.
|
|
"""
|
|
# Open the image, add alpha channel for transparency
|
|
initial_image = Image.open(filename).convert("RGBA")
|
|
|
|
# Check for size
|
|
if initial_image.size[0] != 1304:
|
|
basewidth = 1304
|
|
wpercent = basewidth / float(initial_image.size[0])
|
|
hsize = int((float(initial_image.size[1]) * float(wpercent)))
|
|
initial_image = initial_image.resize((basewidth, hsize), Image.ANTIALIAS)
|
|
# initial_image.save(f"{output_path}/{filename.split('/')[-1]}")
|
|
# raise SystemExit
|
|
|
|
# Convert to numpy array
|
|
initial_image_array = numpy.asarray(initial_image)
|
|
|
|
# Create mask polygon using points of the card border
|
|
mask_polygon = [
|
|
(788, 68),
|
|
(788, 568),
|
|
(515, 568),
|
|
(515, 130),
|
|
(495, 112),
|
|
(491, 104),
|
|
(491, 96),
|
|
(495, 88),
|
|
(515, 72),
|
|
(515, 68),
|
|
(500, 46),
|
|
(499, 46),
|
|
(499, 43),
|
|
(502, 43),
|
|
(512, 47),
|
|
(523, 49),
|
|
(534, 51),
|
|
(550, 52),
|
|
(567, 52),
|
|
(584, 53),
|
|
(719, 53),
|
|
(736, 52),
|
|
(753, 52),
|
|
(769, 51),
|
|
(774, 50),
|
|
(779, 49),
|
|
(785, 48),
|
|
(790, 47),
|
|
(794, 46),
|
|
(803, 43),
|
|
]
|
|
|
|
# Create the mask
|
|
mask_image = Image.new(
|
|
"L", (initial_image_array.shape[1], initial_image_array.shape[0]), 0
|
|
)
|
|
ImageDraw.Draw(mask_image).polygon(mask_polygon, outline=1, fill=1)
|
|
mask = numpy.array(mask_image)
|
|
|
|
# create a new empty image
|
|
extracted_image_array = numpy.empty(initial_image_array.shape, dtype="uint8")
|
|
|
|
# copy the colours from the first 3 columns
|
|
extracted_image_array[:, :, :3] = initial_image_array[:, :, :3]
|
|
|
|
# apply transparency to the alpha channel (4th column)
|
|
extracted_image_array[:, :, 3] = mask * 255
|
|
|
|
# convert back to an image
|
|
extracted_image = Image.fromarray(extracted_image_array, "RGBA")
|
|
|
|
# crop
|
|
extracted_image = extracted_image.crop((490, 42, 803, 568))
|
|
|
|
# save the image
|
|
extracted_image.save(f"{output_path}/{filename.split('/')[-1].split('.')[-2]}.png")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
for image in glob.glob(
|
|
"/Users/dtomlinson/git-repos/web-dev/onmyoji-deck-builder/."
|
|
"image-extraction/images/fields/*.*"
|
|
):
|
|
print(image)
|
|
extract_image(
|
|
image,
|
|
"/Users/dtomlinson/git-repos/web-dev/onmyoji-deck-builder/."
|
|
"image-extraction/images/fields/out",
|
|
)
|