def process_image()

in build_obelics/07_01_nsfw_image_filtering.py [0:0]

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

    def process_image(self, image_bytes):
        try:
            image = Image.open(BytesIO(image_bytes))
            if image.mode != "RGB":
                image = image.convert("RGB")
            if image.size != (IMAGE_DIM, IMAGE_DIM):
                image = image.resize((IMAGE_DIM, IMAGE_DIM))
            image = keras.preprocessing.image.img_to_array(image)
            image /= 255
            return image
        except Exception:
            return keras.preprocessing.image.img_to_array(Image.new("RGB", (IMAGE_DIM, IMAGE_DIM))) / 255