Yasuhito Nagatomo
2 years ago
parent
9543b2d742
commit
9541b078eb
@ -0,0 +1,6 @@
|
||||
{
|
||||
"info" : {
|
||||
"author" : "xcode",
|
||||
"version" : 1
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,12 @@
|
||||
{
|
||||
"images" : [
|
||||
{
|
||||
"filename" : "sample1_512x512.png",
|
||||
"idiom" : "universal"
|
||||
}
|
||||
],
|
||||
"info" : {
|
||||
"author" : "xcode",
|
||||
"version" : 1
|
||||
}
|
||||
}
|
||||
|
After Width: | Height: | Size: 555 KiB |
@ -0,0 +1,78 @@
|
||||
//
|
||||
// ImageToImageView.swift
|
||||
// imggensd2
|
||||
//
|
||||
// Created by Yasuhito Nagatomo on 2023/02/11.
|
||||
//
|
||||
|
||||
import SwiftUI
|
||||
|
||||
struct ImageToImageView: View {
|
||||
static let prompt = "happy smile snow winter"
|
||||
static let negativePrompt =
|
||||
"""
|
||||
lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits,
|
||||
cropped, worst quality, low quality, normal quality, jpeg artifacts, blurry, multiple legs, malformation
|
||||
"""
|
||||
static let startImageName = "sample1_512x512"
|
||||
|
||||
@ObservedObject var imageGenerator: ImageGenerator
|
||||
@State private var generationParameter =
|
||||
ImageGenerator.GenerationParameter(mode: .imageToImage,
|
||||
prompt: prompt,
|
||||
negativePrompt: negativePrompt,
|
||||
guidanceScale: 8.0,
|
||||
seed: 1_000_000,
|
||||
stepCount: 20,
|
||||
imageCount: 1, disableSafety: false,
|
||||
startImage: UIImage(named: startImageName)?.cgImage,
|
||||
strength: 0.5)
|
||||
|
||||
var body: some View {
|
||||
ScrollView {
|
||||
VStack {
|
||||
Text("Image to image").font(.title3).bold().padding(6)
|
||||
Text("Sample App using apple/ml-stable-diffusion")
|
||||
.foregroundColor(.secondary)
|
||||
.font(.caption)
|
||||
.padding(.bottom)
|
||||
|
||||
Image(ImageToImageView.startImageName)
|
||||
.resizable()
|
||||
.scaledToFit()
|
||||
.frame(height: 200)
|
||||
|
||||
PromptView(parameter: $generationParameter)
|
||||
.disabled(imageGenerator.generationState != .idle)
|
||||
|
||||
if imageGenerator.generationState == .idle {
|
||||
Button(action: generate) {
|
||||
Text("Generate").font(.title)
|
||||
}.buttonStyle(.borderedProminent)
|
||||
} else {
|
||||
ProgressView()
|
||||
}
|
||||
|
||||
if let generatedImages = imageGenerator.generatedImages {
|
||||
ForEach(generatedImages.images) {
|
||||
Image(uiImage: $0.uiImage)
|
||||
.resizable()
|
||||
.scaledToFit()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
.padding()
|
||||
}
|
||||
|
||||
func generate() {
|
||||
imageGenerator.generateImages(generationParameter)
|
||||
}
|
||||
}
|
||||
|
||||
struct ImageToImageView_Previews: PreviewProvider {
|
||||
static let imageGenerator = ImageGenerator()
|
||||
static var previews: some View {
|
||||
ImageToImageView(imageGenerator: imageGenerator)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,67 @@
|
||||
//
|
||||
// PromptView.swift
|
||||
// imggensd2
|
||||
//
|
||||
// Created by Yasuhito Nagatomo on 2023/02/11.
|
||||
//
|
||||
|
||||
import SwiftUI
|
||||
|
||||
struct PromptView: View {
|
||||
@Binding var parameter: ImageGenerator.GenerationParameter
|
||||
|
||||
var body: some View {
|
||||
VStack {
|
||||
HStack { Text("Prompt:"); Spacer() }
|
||||
TextField("Prompt:", text: $parameter.prompt)
|
||||
.textFieldStyle(RoundedBorderTextFieldStyle())
|
||||
HStack { Text("Negative Prompt:"); Spacer() }
|
||||
TextField("Negative Prompt:", text: $parameter.negativePrompt)
|
||||
.textFieldStyle(RoundedBorderTextFieldStyle())
|
||||
Stepper(value: $parameter.guidanceScale, in: 0.0...40.0, step: 0.5) {
|
||||
Text("Guidance scale: \(parameter.guidanceScale, specifier: "%.1f") ")
|
||||
}
|
||||
Stepper(value: $parameter.imageCount, in: 1...10) {
|
||||
Text("Image Count: \(parameter.imageCount)")
|
||||
}
|
||||
Stepper(value: $parameter.stepCount, in: 1...100) {
|
||||
Text("Iteration steps: \(parameter.stepCount)")
|
||||
}
|
||||
HStack { Text("Seed:"); Spacer() }
|
||||
TextField("Seed number (0 ... 4_294_967_295)",
|
||||
value: $parameter.seed,
|
||||
formatter: NumberFormatter())
|
||||
.textFieldStyle(RoundedBorderTextFieldStyle())
|
||||
.onSubmit {
|
||||
if parameter.seed < 0 {
|
||||
parameter.seed = 0
|
||||
} else if parameter.seed > UInt32.max {
|
||||
parameter.seed = Int(UInt32.max)
|
||||
} else {
|
||||
// do nothing
|
||||
}
|
||||
}
|
||||
if parameter.mode == .imageToImage {
|
||||
Stepper(value: $parameter.strength, in: 0.0...0.9, step: 0.1) {
|
||||
Text("Strength: \(parameter.strength, specifier: "%.1f") ")
|
||||
}
|
||||
}
|
||||
}
|
||||
.padding()
|
||||
}
|
||||
}
|
||||
|
||||
struct PromptView_Previews: PreviewProvider {
|
||||
@State static var param = ImageGenerator.GenerationParameter(mode: .imageToImage,
|
||||
prompt: "a prompt",
|
||||
negativePrompt: "a negative prompt",
|
||||
guidanceScale: 0.5,
|
||||
seed: 1_000,
|
||||
stepCount: 20,
|
||||
imageCount: 1,
|
||||
disableSafety: false,
|
||||
strength: 0.5)
|
||||
static var previews: some View {
|
||||
PromptView(parameter: $param)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,69 @@
|
||||
//
|
||||
// TextToImageView.swift
|
||||
// imggensd2
|
||||
//
|
||||
// Created by Yasuhito Nagatomo on 2023/02/11.
|
||||
//
|
||||
|
||||
import SwiftUI
|
||||
|
||||
struct TextToImageView: View {
|
||||
static let prompt = "a photo of an astronaut riding a horse on mars"
|
||||
static let negativePrompt =
|
||||
"""
|
||||
lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits,
|
||||
cropped, worst quality, low quality, normal quality, jpeg artifacts, blurry, multiple legs, malformation
|
||||
"""
|
||||
|
||||
@ObservedObject var imageGenerator: ImageGenerator
|
||||
@State private var generationParameter =
|
||||
ImageGenerator.GenerationParameter(mode: .textToImage,
|
||||
prompt: prompt,
|
||||
negativePrompt: negativePrompt,
|
||||
guidanceScale: 8.0,
|
||||
seed: 1_000_000,
|
||||
stepCount: 20,
|
||||
imageCount: 1, disableSafety: false)
|
||||
var body: some View {
|
||||
ScrollView {
|
||||
VStack {
|
||||
Text("Text to image").font(.title3).bold().padding(6)
|
||||
Text("Sample App using apple/ml-stable-diffusion")
|
||||
.foregroundColor(.secondary)
|
||||
.font(.caption)
|
||||
.padding(.bottom)
|
||||
|
||||
PromptView(parameter: $generationParameter)
|
||||
.disabled(imageGenerator.generationState != .idle)
|
||||
|
||||
if imageGenerator.generationState == .idle {
|
||||
Button(action: generate) {
|
||||
Text("Generate").font(.title)
|
||||
}.buttonStyle(.borderedProminent)
|
||||
} else {
|
||||
ProgressView()
|
||||
}
|
||||
|
||||
if let generatedImages = imageGenerator.generatedImages {
|
||||
ForEach(generatedImages.images) {
|
||||
Image(uiImage: $0.uiImage)
|
||||
.resizable()
|
||||
.scaledToFit()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
.padding()
|
||||
}
|
||||
|
||||
func generate() {
|
||||
imageGenerator.generateImages(generationParameter)
|
||||
}
|
||||
}
|
||||
|
||||
struct TextToImageView_Previews: PreviewProvider {
|
||||
static let imageGenerator = ImageGenerator()
|
||||
static var previews: some View {
|
||||
TextToImageView(imageGenerator: imageGenerator)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue