• Home
• com.microsoft.ml.spark
• mmlspark_2.11
• 0.18.1
• source code
• ImageSchemaUtils.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.microsoft.ml.spark.core.schema.ImageSchemaUtils.scala Maven / Gradle / Ivy

// Copyright (C) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See LICENSE in project root for information.

package com.microsoft.ml.spark.core.schema

import org.apache.spark.ml.image.ImageSchema
import org.apache.spark.sql.types._

object ImageSchemaUtils {

  val ColumnSchemaNullable = {
    StructType(
      StructField("origin", StringType, true) ::
        StructField("height", IntegerType, true) ::
        StructField("width", IntegerType, true) ::
        StructField("nChannels", IntegerType, true) ::
        // OpenCV-compatible type: CV_8UC3 in most cases
        StructField("mode", IntegerType, true) ::
        // Bytes in OpenCV-compatible order: row-wise BGR in most cases
        StructField("data", BinaryType, true) :: Nil)
  }

  val ImageSchemaNullable = StructType(StructField("image", ColumnSchemaNullable, true) :: Nil)

  def isImage(dataType: DataType): Boolean = {
    dataType == ImageSchema.columnSchema ||
      dataType == ColumnSchemaNullable
  }

  def isImage(dataType: StructField): Boolean = {
    isImage(dataType.dataType)
  }
}