kunofellasleep

kernel void sharpen(
                   texture2d<float, access::write> outTex [[texture(0)]],
                   texture2d<float, access::read> inTex [[texture(1)]],
                   uint2 gid [[thread_position_in_grid]]) {
    float4 c = float4(0.0, 0.0, 0.0, 1.0);
    //隣接ピクセルを使用して色を決定するためのパラメータ
    float weight[3][3] = {
        {0, -1, 0},
        {-1,  5, -1},
        {0, -1, 0}

kunofellasleep

import Metal
import MetalKit

class ImageProcessor: NSObject {
    
    //ハードウェアとしてのGPUを抽象化したプロトコル
    lazy var device: MTLDevice! = MTLCreateSystemDefaultDevice()
    //コマンドバッファの実行順を管理するキュー
    var commandQueue: MTLCommandQueue!

kunofellasleep

func image(from mtlTexture: MTLTexture) -> UIImage {
    //画像サイズ
    let w = mtlTexture.width
    let h = mtlTexture.height
    let bytesPerPixel: Int = 4
    let imageByteCount = w * h * bytesPerPixel
    let bytesPerRow = w * bytesPerPixel
    var src = [UInt8](repeating: 0, count: Int(imageByteCount))
    //CGImageに変換
    let region = MTLRegionMake2D(0, 0, w, h)

kunofellasleep

public func Run(_ image:UIImage) -> UIImage{
    //GPUで実行されるコマンドを格納するコンテナ
    let buffer = commandQueue.makeCommandBuffer()
    //コマンドを作成し、コマンドバッファに追加する（エンコード）
    let encoder = buffer?.makeComputeCommandEncoder()
    encoder?.setComputePipelineState(pipelineState)
    //出力用テクスチャ作成
    let textureDescriptor = MTLTextureDescriptor.texture2DDescriptor(pixelFormat: MTLPixelFormat.rgba8Unorm, width:1920, height: 1080, mipmapped: false)
    textureDescriptor.usage = [.shaderRead, .shaderWrite]
    outTexture = self.device.makeTexture(descriptor: textureDescriptor) 

kunofellasleep

func mtlTexture(from image: UIImage) -> MTLTexture {
    //CGImage変換時に向きがおかしくならないように
    UIGraphicsBeginImageContext(image.size);
    image.draw(in: CGRect(x:0, y:0, width:image.size.width, height:image.size.height))
    let orientationImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    //CGImageに変換
    guard let cgImage = orientationImage?.cgImage else {
        fatalError("Can't open image \(image)")
    }

kunofellasleep

//ハードウェアとしてのGPUを抽象化したプロトコル
lazy var device: MTLDevice! = MTLCreateSystemDefaultDevice()
//コマンドバッファの実行順を管理するキュー
var commandQueue: MTLCommandQueue!

public func Setup() {
    let defaultLibrary = device.makeDefaultLibrary()!
    if let target = defaultLibrary.makeFunction(name: funcName) {
        commandQueue = device.makeCommandQueue()
        do {

kunofellasleep

#include <metal_stdlib>
using namespace metal;
kernel void duotone(
  texture2d<float, access::write> outTex [[texture(0)]],
  texture2d<float, access::read> inTex [[texture(1)]],
  uint2 gid [[thread_position_in_grid]])
{
    float3 c = inTex.read(gid).rgb;
    float mn = (c.r + c.g + c.b) / 3.0;
    float4 out = float4(0.5, 0.5 + mn, 1.0, 1.0);

kunofellasleep

python retrain.py --image_dir ./images
python label_image.py --image test.jpg --graph output_graph.pb --labels output_labels.txt --input_layer=Placeholder

kunofellasleep

googleimagesdownload -k keywordA -sk keywordB -f "jpg" -s medium

kunofellasleep

// ref :: http://t-pot.com/program/90_BitonicSort/index.html

#include <metal_stdlib>
using namespace metal;

kernel void bitonicSort(texture2d<float, access::read> inTexture [[texture(0)]],
                    texture2d<float, access::write> outTexture [[texture(1)]],
                    device float *stepno [[buffer(0)]],
                    device float *offset [[buffer(1)]],