MvvmScaffoldFrame48/MvvmScaffoldFrame48.VIEWMODEL/ViewModel/DeepObjectDetectClass.cs

using MvvmScaffoldFrame48.DLL.ImageAlgorithm;
using System;
using System.Collections.Generic;
using System.Drawing;
using System.Drawing.Imaging;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Input;
using System.Windows.Media.Imaging;
using System.Xml.Linq;
using VisionDesigner.MVDCNNDetect;

namespace MvvmScaffoldFrame48.ViewModel.ViewModel
{
    public class DeepObjectDetectClass : BaseViewModel
    {
        #region 界面绑定属性
        private BitmapImage _showImage;
        public BitmapImage ShowImage 
        {
            get { return _showImage; }
            set
            {
                if (_showImage != value)
                {
                    _showImage = value;
                    OnPropertyChanged(nameof(ShowImage));
                }
            }
        }
        #endregion

        #region 界面绑定事件
        public ICommand TestCommand{ get; set; }
        #endregion

        #region 属性
        private CPUDeepObjectDetectClass cPUDeepObject = new CPUDeepObjectDetectClass();
        #endregion

        #region 绑定用Action方法
        public void Test(object obj)
        {
            List<Bitmap> bitmaps = new List<Bitmap>();
            // 创建一个 Bitmap 对象
            Bitmap bitmap = new Bitmap("D:\\work\\MvvmScaffoldFrame48\\MvvmScaffoldFrame48\\MvvmScaffoldFrame48\\testfile\\testimage\\2025-08-04-13-28-45-807.jpg");

            cPUDeepObject.LoadImage(bitmap);
            cPUDeepObject.LoadDeepModel("D:\\work\\MvvmScaffoldFrame48\\MvvmScaffoldFrame48\\MvvmScaffoldFrame48\\testfile\\model\\Detect_20250818T1328175385.bin");
            var result = cPUDeepObject.RunDetect();
            foreach (var item in result) 
            {
                if (item.LabelName == "A好药")
                {
                    continue;
                }
                Rectangle rectangle = new Rectangle(Convert.ToInt32(item.Box.CenterX - (item.Box.Width / 2)), Convert.ToInt32(item.Box.CenterY - (item.Box.Height / 2)), Convert.ToInt32(item.Box.Width), Convert.ToInt32(item.Box.Height));
                bitmaps.Add(ImageAlgorithmTools.CropBitmap(bitmap, rectangle));
            }
            //PrintRectToBitmap(result, ref bitmap);
            // 将 Bitmap 转换为 BitmapImage
            //BitmapImage bitmapImage = ImageAlgorithmTools.ConvertToBitmapImage(bitmap);
            //ShowImage = bitmapImage;

            Task.Run(async () =>
            {
                foreach (var item in bitmaps)
                {
                    await Application.Current.Dispatcher.InvokeAsync(() => {
                        BitmapImage bitmapImage = ImageAlgorithmTools.ConvertToBitmapImage(item);
                        ShowImage = bitmapImage;
                    });
                    Thread.Sleep(500);
                }
            });
        }
        #endregion

        #region 绑定用Predicate方法
        private bool CanTrue(object obj)
        {
            return true;
        }
        private bool CanFalse(object obj)
        {
            return false;
        }
        #endregion

        #region 其他方法

        public DeepObjectDetectClass()
        {
            TestCommand = new RelayCommand(Test, CanTrue);
        }

        public void PrintRectToBitmap(List<CNNDetectPredInfo> RectData , ref Bitmap bitmap)
        {
            using (Graphics g = Graphics.FromImage(bitmap))
            {
                var redPen = new Pen(Color.Red, 2);
                var GreenPen = new Pen(Color.SeaGreen, 2);

                foreach (var item in RectData)
                {
                    int roix = Convert.ToInt32(item.Box.CenterX - (item.Box.Width/2));
                    int roiy =  Convert.ToInt32(item.Box.CenterY - (item.Box.Height/2));
                    g.DrawRectangle(item.LabelName== "A坏药"?redPen:GreenPen, new Rectangle(roix, roiy, (int)item.Box.Width, (int)item.Box.Height));
                }
            }
        }
        #endregion
    }
}