Jan 17

Beautiful inpainting How to use Pytorch-MADF

Top quality model

Inpainting

2 min read

Inpainting

2 min read

Jan 16

Use generative-inpainting-pytorch

Method install git clone https://github.com/daa233/generative-inpainting-pytorch.git cd generative-inpainting-pytorch pip install pyyaml==5.4.1 Inference python test_single.py --checkpoint_path path_to_pretrained_model_dir --image your_image.png --mask examples/center_mask_256.png --output examples/output.png

Inpainting

1 min read

Inpainting

1 min read

Use generative-inpainting-pytorch

Method

install

git clone https://github.com/daa233/generative-inpainting-pytorch.git
cd generative-inpainting-pytorch
pip install pyyaml==5.4.1

Inference

python test_single.py --checkpoint_path path_to_pretrained_model_dir --image your_image.png --mask examples/center_mask_256.png --output examples/output.png

Jan 16

How to use Yolov6

Exceeding v7 v8? ? GitHub - meituan/YOLOv6: YOLOv6: a single-stage object detection framework dedicated to industrial… YOLOv6: a single-stage object detection framework dedicated to industrial applications. - GitHub - meituan/YOLOv6…github.com According to Yolov6 repository, it is faster and more accurate than v7 and v8. Usage install git clone https://github.com/meituan/YOLOv6.git cd YOLOv6 pip install -r requirements.txt wget https://github.com/meituan/YOLOv6/releases/download/0.3.0/yolov6s.pt # Download pre-trained model.

Yolov 6

2 min read

Yolov 6

2 min read

Jan 15

Convert multi-input Pytorch model to CoreML model

Handle models with multiple inputs Some machine learning models require multiple inputs. How do weconvert such models to CoreML? Method Suppose we transform a model with two inputs, x and y. import torch import coremltools as ct traced_model = torch.jit.trace(model, (x,y)) # convert to torchscript mlmodel = ct.convert(traced_model, inputs=[ct.TensorType(shape=x.shape),ct.TensorType(shape=y.shape)]) # convert to coreml

Coreml

1 min read

Coreml

1 min read

Convert multi-input Pytorch model to CoreML model

Handle models with multiple inputs

Some machine learning models require multiple inputs.
How do weconvert such models to CoreML?

Method

Suppose we transform a model with two inputs, x and y.

import torch
import coremltools as ct

traced_model = torch.jit.trace(model, (x,y)) # convert to torchscript
mlmodel = ct.convert(traced_model, inputs=[ct.TensorType(shape=x.shape),ct.TensorType(shape=y.shape)]) # convert to coreml

Usage in Swift

Run the converted model in Swift.

do {
    let model = try myModel().model
    let input = myModelInput(x: x, y: y) // x and y are MLMultiArray in this case.
    let out = try model.prediction(from: input)
} catch let error {
    print(error)
}

🐣

I’m a freelance engineer.
Work consultation
Please feel free to contact us with a brief development description.
rockyshikoku@gmail.com

I am making an app that uses Core ML and ARKit.
We send machine learning / AR related information.

GitHub

Twitter
Medium

Jan 12

UIImage to MLMultiArray

When you want to run a CoreML model whose input is TensorType CoreML models with ImageType input can use CIImage, CGImage, and PixelBuffer as inputs, but CoreML models with TensorType input require MLMultiArray as input. How to convert from UIImage to MLMultiArray when you want to use image input for such model. Method It can be converted with the following UIImageExtension. // Usage: //…

Swif

3 min read

Swif

3 min read

Jan 11

How to use Yolov8

A new model for object detection. Easier to use. GitHub - ultralytics/ultralytics: YOLOv8 🚀 in PyTorch > ONNX > CoreML > TFLite Ultralytics YOLOv8, developed by Ultralytics, is a cutting-edge, state-of-the-art (SOTA) model that builds upon the…github.com Usage Install pip install ultralytics Predict Object Detection yolo task=detect mode=predict model=yolov8s.pt source="https://ultralytics.com/images/bus.jpg"

Yolov 8

3 min read

Yolov 8

3 min read

Aug 8, 2022

Solution for “PyTorch convert function for op ‘_record_function_enter’ not implemented.” in CoreMLTools

I get an error and can’t convert (cries) When converting a Pytorch model with CoreMLTools, you may encounter the title error. This model cannot be converted because the necessary operations are not available in CoreMLTools. . . Solvable If you think like that, you can actually solve it easily. The model with the title error is torch.nn.DataParallel(module, device_ids=None, output_device=None…

Python

1 min read

Python

1 min read

Aug 5, 2022

Use Blind Super Resolution model DCLS-SR to super-resolve image degradation

I want to clean a degraded image [Image: input→output]

Python

2 min read

Use Blind Super Resolution model DCLS-SR to super-resolve image degradation

Python

2 min read

Aug 4, 2022

Convert exponential notation “e+” to decimal notation in Pytorch

Exponential notation is confusing When you print torch.tensor, it becomes exponential notation “+e”, and the meaning is sometimes difficult to understand. I want to convert it to normal decimal notation. torch.set_printoptions This will give you decimal notation. 🐣 I’m a freelance engineer. Work consultation Please feel free to contact us with a…

Python

1 min read

Python

1 min read