Ejemplo n.º 1
0
        public static void TestFilterGroundTruthFile()
        {
            string outputDir = DirectoryConstants.defaultTempDirectory + "filteredKITTIGroundTruth\\";
            string outputDir_Seattle = outputDir + "SeattleLive-5-WestLake-NS\\";
            string outputDir_Stewart = outputDir + "1-Stewart-NS\\";
            string outputDir_BellevueNE = outputDir + "Bellevue-148th-NE29th\\";
            string outputDir_BellevueMain = outputDir + "Bellevue-148th-Main\\";

            string inputDir = @"C:\research\dataset\LongDuration\";
            string inputDir_Seattle = inputDir + "SeattleLive-5-WestLake-NS\\";
            string inputDir_Stewart = inputDir + "1-Stewart-NS\\";
            string inputDir_BellevueNE = inputDir + "Bellevue-148th-NE29th\\";
            string inputDir_BellevueMain = inputDir + "Bellevue-148th-Main\\";

            List<string> inputs = new List<string>()
            {
                inputDir_BellevueNE,
                inputDir_BellevueMain,
                inputDir_Seattle,
                inputDir_Stewart,
            };
            List<string> outputs = new List<string>()
            {
                outputDir_BellevueNE,
                outputDir_BellevueMain,
                outputDir_Seattle,
                outputDir_Stewart,
            };


            for (int i = 0; i < inputs.Count; i++)
            {
                string[] subdirs = Directory.GetDirectories(inputs[i] + "labels\\");
                foreach (string subdir in subdirs)
                {
                    int omitted = 0;
                    string[] files = Directory.GetFiles(subdir);
                    if (files.Length == 0)
                    {
                        string [] reward = Directory.GetDirectories(subdir);
                        files = Directory.GetFiles(reward[0]);
                    }
                    string groupName = URIUtilities.localDirectoryNameFromDirectory(subdir);
                    foreach (string file in files)
                    {
                        int tmp;
                        KITTIDetectionResultValidation.FilterGroundTruthFile(file, outputs[i] + groupName, out tmp);
                        omitted += tmp;
                    }
                    Console.WriteLine("{0}: omitted {1}", subdir, omitted);
                }
                
            }
        }