コード例 #1
0
        private async Task RunAsync()
        {
            var modelConfig       = ModelConfig.Value;
            var cancellationToken = Lifetime.ApplicationStopped;

            var loading = new GraphLoading(modelConfig.ModelToUse);
            var catalog = new Catalog(modelConfig.LabelsToUse);
            await catalog.LoadAsync(cancellationToken);

            using (var resultBuilder = new ResultBuilder(modelConfig.OutputFileName, catalog))
                using (var graph = await loading.LoadGraphAsync(cancellationToken))
                    using (var session = new TFSession(graph))
                    {
                        var batch = new List <string>(modelConfig.BatchSize);

                        foreach (var imageDirectory in modelConfig.ImageDirectories)
                        {
                            foreach (var inputFile in Directory.EnumerateFiles(imageDirectory, "*.jpg", SearchOption.AllDirectories))
                            {
                                cancellationToken.ThrowIfCancellationRequested();
                                batch.Add(Path.Combine(imageDirectory, inputFile));
                                if (batch.Count == batch.Capacity)
                                {
                                    await ProcessBatch(resultBuilder, graph, session, batch, cancellationToken);

                                    batch.Clear();
                                }
                            }
                        }
                    }
            Logger.LogInformation("All images processed");
            Lifetime.StopApplication();
        }
コード例 #2
0
ファイル: Program.cs プロジェクト: nrandell/image-monitor
        private static async Task RunAsync(string[] args, CancellationToken ct)
        {
            var loading = new GraphLoading(Path.Combine(ModelDirectory, "frozen_inference_graph.pb"));
            var catalog = new Catalog(CatalogFile);

            await catalog.LoadAsync(ct);

            var graph = await loading.LoadGraphAsync(ct);

            using (var session = new TFSession(graph))
                using (var input = new InputData(ImageSize, BatchSize))
                {
                    foreach (var directoryName in args)
                    {
                        var imageFiles = Directory.EnumerateFiles(directoryName, "*.jpg", SearchOption.AllDirectories);
                        var batch      = new List <string>(BatchSize);
                        foreach (var imageFile in imageFiles)
                        {
                            batch.Add(Path.Combine(directoryName, imageFile));
                            if (batch.Count == BatchSize)
                            {
                                await TryProcessImage(catalog, graph, session, input, batch, ct);

                                batch.Clear();
                            }
                        }
                        if (batch.Count > 0)
                        {
                            await TryProcessImage(catalog, graph, session, input, batch, ct);
                        }
                    }
                }
        }