xcode - MLModel 如何分析 ImagePicker?
问题描述
我仍然对如何制作由机器学习模型文件识别的当前图像(从图库照片中选择)感到困惑。我想做的是,一旦点击相机按钮,我就可以从图库图片中选择一张图片(这是代码的一部分)。但是,一旦显示照片,它不会触发 ml 模型,并且照片不会被识别,而是按原样显示。有人可以帮我吗?这是我的代码:
import SwiftUI
struct ContentView: View {
@State private var isShowPicker: Bool = false
@State private var image: Image? = Image("")
@State private var classificationLabel: String = ""
let model = MobileNetV2()
private func performImageClassification() {
let currentImageName = ImagePicker(image: self.$image)
guard let img = UIImage(named: "\(String(describing: currentImageName))"),
let resizedImage = img.resizeTo(size: CGSize(width: 224, height: 224)),
let pixelBuffer = resizedImage.toCVPixelBuffer() else {
return
}
let output = try? model.prediction(image: pixelBuffer)
if let output = output {
self.classificationLabel = output.classLabel
}
}
var body: some View {
NavigationView {
ZStack {
VStack {
image?
.resizable()
.navigationBarItems(trailing:
Button(action: {
self.performImageClassification()
self.isShowPicker.toggle()
})
{
Image(systemName: "camera.fill")
.font(.largeTitle)
}.foregroundColor(.blue)
).navigationBarBackButtonHidden(false)
.navigationBarTitle("\(classificationLabel)")
Text(classificationLabel)
Spacer()
}
}
.sheet(isPresented: $isShowPicker) {
ImagePicker(image: self.$image)
}
.navigationBarTitle(Text(classificationLabel))
}
}
}
struct ImagePicker: UIViewControllerRepresentable {
@Environment(\.presentationMode)
var presentationMode
@Binding var image: Image?
class Coordinator: NSObject, UINavigationControllerDelegate, UIImagePickerControllerDelegate {
@Binding var presentationMode: PresentationMode
@Binding var image: Image?
init(presentationMode: Binding<PresentationMode>, image: Binding<Image?>) {
_presentationMode = presentationMode
_image = image
}
func imagePickerController(_ picker: UIImagePickerController,
didFinishPickingMediaWithInfo info: [UIImagePickerController.InfoKey : Any]) {
let uiImage = info[UIImagePickerController.InfoKey.originalImage] as! UIImage
image = Image(uiImage: uiImage)
presentationMode.dismiss()
}
func imagePickerControllerDidCancel(_ picker: UIImagePickerController) {
presentationMode.dismiss()
}
}
func makeCoordinator() -> Coordinator {
return Coordinator(presentationMode: presentationMode, image: $image)
}
func makeUIViewController(context: UIViewControllerRepresentableContext<ImagePicker>) -> UIImagePickerController {
let picker = UIImagePickerController()
picker.delegate = context.coordinator
return picker
}
func updateUIViewController(_ uiViewController: UIImagePickerController,
context: UIViewControllerRepresentableContext<ImagePicker>) {
}
}
struct ContentView_Previews: PreviewProvider {
static var previews: some View {
NavigationView {
ContentView()
}
}
}
解决方案
这将起作用:
import SwiftUI
class ImagePickerCoordinator: NSObject, UINavigationControllerDelegate, UIImagePickerControllerDelegate {
@Binding var image: UIImage?
@Binding var isShown: Bool
init(image: Binding<UIImage?>,
isShown: Binding<Bool>) {
_image = image
_isShown = isShown
}
func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [UIImagePickerController.InfoKey : Any]) {
if let uiImage = info[UIImagePickerController.InfoKey.originalImage] as? UIImage {
image = uiImage
isShown = false
}
}
func imagePickerControllerDidCancel(_ picker: UIImagePickerController) {
isShown = false
}
}
struct ImagePicker: UIViewControllerRepresentable {
typealias UIViewControllerType = UIImagePickerController
typealias Coordinator = ImagePickerCoordinator
@Binding var image: UIImage?
@Binding var isShown: Bool
var sourceType: UIImagePickerController.SourceType = .camera
func updateUIViewController(_ uiViewController: UIImagePickerController, context: UIViewControllerRepresentableContext<ImagePicker>) {
}
func makeCoordinator() -> ImagePicker.Coordinator {
ImagePickerCoordinator(image: $image, isShown: $isShown)
}
func makeUIViewController(context: UIViewControllerRepresentableContext<ImagePicker>) -> UIImagePickerController {
let picker = UIImagePickerController()
picker.sourceType = sourceType
picker.delegate = context.coordinator
return picker
}
}
推荐阅读
- reactjs - 将选定的选项传递给父组件 - 反应
- ios - 使用 CABasicAnimation 缩放 CAShapeLayer 会导致它平移
- android - 我可以在我的 Flutter 应用中找到 Android 资源名称吗?
- go - GORM 逗号分隔值字段
- api - 尝试通过 API 创建付款时出错
- r - 将聚合函数应用于具有自定义输出名称的多个 data.table 列
- rust - 'cargo run' 和启动 .exe 文件之间的区别
- angular - 如何在 p-card 中设置固定宽度
- ruby-on-rails - 是否可以覆盖 SimpleForm 生成的表单字段的基“类”?
- python - 基于字典和 str.contains() 的新数据框列