当前位置:   article > 正文

微软AI系列 C#中实现相似度计算涉及到加载图像、使用预训练的模型提取特征以及计算相似度_c# 搭建ai训练模型

c# 搭建ai训练模型

在C#中实现相似度计算涉及到加载图像、使用预训练的模型提取特征以及计算相似度。你可以使用.NET中的深度学习库如TensorFlow.NET来加载预训练模型,提取特征,并进行相似度计算。

以下是一个使用TensorFlow.NET的示例:

  1. using System;
  2. using TensorFlow;
  3. using TensorFlow.Image;
  4. class Program
  5. {
  6. static void Main(string[] args)
  7. {
  8. // 载入模型
  9. var model = new ResNet50();
  10. // 加载图像
  11. var image1 = ImageUtil.LoadTensorFromImageFile("image1.jpg");
  12. var image2 = ImageUtil.LoadTensorFromImageFile("image2.jpg");
  13. // 提取特征
  14. var feature1 = ExtractFeatures(image1, model);
  15. var feature2 = ExtractFeatures(image2, model);
  16. // 计算相似度
  17. var similarityScore = CalculateSimilarity(feature1, feature2);
  18. Console.WriteLine("图片相似度: " + similarityScore);
  19. }
  20. static TFTensor ExtractFeatures(TFTensor image, ResNet50 model)
  21. {
  22. // 预处理图像
  23. var processedImage = ImageUtil.ResizeAndCropCenter(image, model.InputHeight, model.InputWidth);
  24. processedImage = ImageUtil.Normalize(image, mean: model.Mean, std: model.Std);
  25. // 转换图像形状以匹配模型输入
  26. var reshapedImage = processedImage.Reshape(new long[] { 1, model.InputHeight, model.InputWidth, 3 });
  27. // 获取特征
  28. var features = model.Predict(reshapedImage);
  29. return features;
  30. }
  31. static double CalculateSimilarity(TFTensor feature1, TFTensor feature2)
  32. {
  33. // 使用余弦相似度计算特征之间的相似度
  34. var similarity = CosineSimilarity(feature1.ToArray<float>(), feature2.ToArray<float>());
  35. return similarity;
  36. }
  37. static double CosineSimilarity(float[] vector1, float[] vector2)
  38. {
  39. double dotProduct = 0.0;
  40. double magnitude1 = 0.0;
  41. double magnitude2 = 0.0;
  42. for (int i = 0; i < vector1.Length; i++)
  43. {
  44. dotProduct += vector1[i] * vector2[i];
  45. magnitude1 += Math.Pow(vector1[i], 2);
  46. magnitude2 += Math.Pow(vector2[i], 2);
  47. }
  48. magnitude1 = Math.Sqrt(magnitude1);
  49. magnitude2 = Math.Sqrt(magnitude2);
  50. return dotProduct / (magnitude1 * magnitude2);
  51. }
  52. }

在这个示例中,我们使用了TensorFlow.NET库中的ResNet50模型来提取图像的特征表示。我们首先载入模型,然后加载图片并对其进行预处理,接着提取特征,并最后使用余弦相似度计算图片的相似度。

请确保在项目中包含了TensorFlow.NET的引用,并根据实际情况修改图片的路径以及模型的输入参数。

使用Python实现了同样的逻辑,可以对比 参考

  1. import numpy as np
  2. from tensorflow.keras.preprocessing import image
  3. from tensorflow.keras.applications import ResNet50
  4. from tensorflow.keras.applications.resnet50 import preprocess_input
  5. from sklearn.metrics.pairwise import cosine_similarity
  6. # 加载预训练的ResNet50模型
  7. model = ResNet50(weights='imagenet', include_top=False, pooling='avg')
  8. # 加载并预处理图像
  9. def preprocess_image(img_path):
  10. img = image.load_img(img_path, target_size=(224, 224))
  11. x = image.img_to_array(img)
  12. x = np.expand_dims(x, axis=0)
  13. x = preprocess_input(x)
  14. return x
  15. # 提取图像的特征向量
  16. def extract_features(img_path, model):
  17. img = preprocess_image(img_path)
  18. features = model.predict(img)
  19. return features.flatten()
  20. # 图像路径
  21. image1_path = '/Users/AI/pythonsamples-main/ML/CNN(卷积神经网络)/ImageVector/houge.jpg'
  22. image2_path = '/Users/AI/pythonsamples-main/ML/CNN(卷积神经网络)/ImageVector/zhipiao.jpg'
  23. # 提取特征向量
  24. features1 = extract_features(image1_path, model)
  25. features2 = extract_features(image2_path, model)
  26. # 计算余弦相似度
  27. similarity = cosine_similarity([features1], [features2])[0][0]
  28. print("相似度:", similarity)

结果:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/941386
推荐阅读
相关标签
  

闽ICP备14008679号