Ejemplo n.º 1
0
        public bool LoadCapture(int batchcount, int inchannels, int inw, int inh, int outchannels, int outw, int outh, bool doFlipX, bool doFlipY, double doRot, bool doAffine, out RNdMatrix smat, out RNdMatrix tmat)
        {
            bool ret = true;

            List <Mat> sframes = new List <Mat>();
            List <Mat> tframes = new List <Mat>();

            for (int i = 0; i < batchcount; i++)
            {
                Mat mat = new Mat(), mat1, mat2;
                while (!capture.Read(mat))
                {
                }
                mat1 = mat.Clone();
                mat2 = mat.Clone();

                bool   flipx = false;
                bool   flipy = false;
                int    oflux = 0, ofluy = 0;
                int    ofrdx = 0, ofrdy = 0;
                double scl = 2.5;
                double rotangle = 0;
                Point  lu = new Point(), rd = new Point();

                if (doFlipX)
                {
                    flipx = (State.RandomSource.NextDouble() > 0.5 ? true : false);
                }
                if (doFlipY)
                {
                    flipy = (State.RandomSource.NextDouble() > 0.5 ? true : false);
                }
                rotangle = (State.RandomSource.NextDouble() * 2 - 1) * doRot;
                if (doAffine)
                {
                    if (inw > 1 && outw > 1 && inh > 1 && outh > 1)
                    {
                        oflux = State.RandomSource.Next(Math.Max(0, (int)(Math.Min(inw, outw) / (scl) - 1)));
                        ofluy = State.RandomSource.Next(Math.Max(0, (int)(Math.Min(inh, outh) / (scl) - 1)));
                        ofrdx = State.RandomSource.Next(Math.Max(0, (int)(Math.Min(inw, outw) / (scl) - 1)));
                        ofrdy = State.RandomSource.Next(Math.Max(0, (int)(Math.Min(inh, outh) / (scl) - 1)));
                    }
                }

                lu = new Point(oflux, ofluy);
                rd = new Point(ofrdx, ofrdy);

                var sframe = mat1.Clone();
                if (inchannels != sframe.Channels())
                {
                    if (inchannels == 1)
                    {
                        Cv2.CvtColor(sframe, sframe, ColorConversionCodes.BGR2GRAY);
                    }
                    else if (inchannels == 3)
                    {
                        Cv2.CvtColor(sframe, sframe, ColorConversionCodes.GRAY2BGR);
                    }
                    else
                    {
                        throw new Exception();
                    }
                }
                sframe = sframe.Resize(new Size(inw, inh), 0, 0, InterpolationFlags.Area);
                sframe = EffectProcess.Offset(sframe, lu, rd, flipx, flipy, rotangle);
                sframes.Add(sframe.Clone());

                var tframe = mat2.Clone();
                if (outchannels != tframe.Channels())
                {
                    if (outchannels == 1)
                    {
                        Cv2.CvtColor(tframe, tframe, ColorConversionCodes.BGR2GRAY);
                    }
                    else if (outchannels == 3)
                    {
                        Cv2.CvtColor(tframe, tframe, ColorConversionCodes.GRAY2BGR);
                    }
                    else
                    {
                        throw new Exception();
                    }
                }
                tframe = EffectProcess.Effect(tframe.Resize(new Size(outw, outh), 0, 0, InterpolationFlags.Area));
                tframe = EffectProcess.Offset(tframe, lu, rd, flipx, flipy, rotangle);
                tframes.Add(tframe.Clone());
            }

            Converter.MatToRNdMatrix(sframes.ToArray(), out smat);
            Converter.MatToRNdMatrix(tframes.ToArray(), out tmat);
            return(ret);
        }
Ejemplo n.º 2
0
        public bool LoadImage(int batchcount, int inchannels, int inw, int inh, int outchannels, int outw, int outh, bool doFlipX, bool doFlipY, double doRot, bool doAffine, out RNdMatrix smat, out RNdMatrix tmat)
        {
            bool ret = false;

            smat = tmat = null;

            List <Mat> sframes = new List <Mat>();
            List <Mat> tframes = new List <Mat>();

            for (int i = 0; i < batchcount; i++)
            {
                bool check;
                var  index = GetIndex(out check, inw, inh);
                ret |= check;
                var mat1 = (index[0].Container);
                var mat2 = (index[1].Container);

                bool   flipx = false;
                bool   flipy = false;
                int    oflux = 0, ofluy = 0;
                int    ofrdx = 0, ofrdy = 0;
                double scl = 2.5;
                double rotangle = 0;
                Point  lu = new Point(), rd = new Point();

                if (doFlipX)
                {
                    flipx = (State.RandomSource.NextDouble() > 0.5 ? true : false);
                }
                if (doFlipY)
                {
                    flipy = (State.RandomSource.NextDouble() > 0.5 ? true : false);
                }
                rotangle = (State.RandomSource.NextDouble() * 2 - 1) * doRot;
                if (doAffine)
                {
                    var w = Math.Min(mat1.Width, mat2.Width);
                    var h = Math.Min(mat1.Height, mat2.Height);
                    {
                        oflux = State.RandomSource.Next(Math.Max(0, (int)(w / (scl) - 1)));
                        ofluy = State.RandomSource.Next(Math.Max(0, (int)(h / (scl) - 1)));
                        ofrdx = State.RandomSource.Next(Math.Max(0, (int)(w / (scl) - 1)));
                        ofrdy = State.RandomSource.Next(Math.Max(0, (int)(h / (scl) - 1)));
                    }
                }

                lu = new Point(oflux, ofluy);
                rd = new Point(ofrdx, ofrdy);

                var sframe = mat1.Clone();
                if (inchannels != sframe.Channels())
                {
                    if (inchannels == 1)
                    {
                        Cv2.CvtColor(sframe, sframe, ColorConversionCodes.BGR2GRAY);
                    }
                    else if (inchannels == 3)
                    {
                        Cv2.CvtColor(sframe, sframe, ColorConversionCodes.GRAY2BGR);
                    }
                    else
                    {
                        throw new Exception();
                    }
                }
                sframe = EffectProcess.Offset(sframe, lu, rd, flipx, flipy, rotangle);
                sframe = sframe.Resize(new Size(inw, inh), 0, 0, InterpolationFlags.Area);
                sframes.Add(sframe.Clone());

                var tframe = mat2.Clone();
                if (outchannels != tframe.Channels())
                {
                    if (outchannels == 1)
                    {
                        Cv2.CvtColor(tframe, tframe, ColorConversionCodes.BGR2GRAY);
                    }
                    else if (outchannels == 3)
                    {
                        Cv2.CvtColor(tframe, tframe, ColorConversionCodes.GRAY2BGR);
                    }
                    else
                    {
                        throw new Exception();
                    }
                }
                tframe = EffectProcess.Effect(tframe);
                tframe = EffectProcess.Offset(tframe, lu, rd, flipx, flipy, rotangle);
                tframe = tframe.Resize(new Size(outw, outh), 0, 0, InterpolationFlags.Area);
                tframes.Add(tframe.Clone());
            }

            Converter.MatToRNdMatrix(sframes.ToArray(), out smat);
            Converter.MatToRNdMatrix(tframes.ToArray(), out tmat);
            return(ret);
        }