Пример #1
0
        private async Task <StorageFile> GetModelFileAsync(CustomVisionModelData customVisionModelData)
        {
            if (customVisionModelData.IsPrebuiltModel)
            {
                string modelPath = $"Assets\\{customVisionModelData.FileName}";
                return(await Package.Current.InstalledLocation.GetFileAsync(modelPath));
            }
            else
            {
                StorageFolder onnxProjectDataFolder = await CustomVisionDataLoader.GetOnnxModelStorageFolderAsync(CustomVisionProjectType.Classification);

                return(await onnxProjectDataFolder.GetFileAsync(customVisionModelData.FileName));
            }
        }
Пример #2
0
        private async Task DeleteProjectAsync()
        {
            CustomVisionModelData currentProject = this.projectsComboBox.SelectedValue as CustomVisionModelData;

            if (currentProject != null && !currentProject.IsPrebuiltModel)
            {
                // delete ONNX model file
                StorageFolder onnxProjectDataFolder = await CustomVisionDataLoader.GetOnnxModelStorageFolderAsync(CustomVisionProjectType.Classification);

                StorageFile modelFile = await onnxProjectDataFolder.GetFileAsync(currentProject.FileName);

                if (modelFile != null)
                {
                    await modelFile.DeleteAsync();
                }

                // update local file with custom models
                this.Projects.Remove(currentProject);
                List <CustomVisionModelData> updatedCustomModelList = this.Projects.Where(x => !x.IsPrebuiltModel).ToList();
                await CustomVisionDataLoader.SaveCustomVisionModelDataAsync(updatedCustomModelList, CustomVisionProjectType.Classification);

                this.projectsComboBox.SelectedIndex = 0;
            }
        }
Пример #3
0
        private async Task <bool> ExportOnnxProject(Export exportProject, CustomVisionProjectType customVisionProjectType)
        {
            if (string.IsNullOrEmpty(exportProject?.DownloadUri))
            {
                throw new ArgumentNullException("Download Uri");
            }

            bool          success               = false;
            Guid          newModelId            = Guid.NewGuid();
            StorageFolder onnxProjectDataFolder = await CustomVisionDataLoader.GetOnnxModelStorageFolderAsync(customVisionProjectType);

            StorageFile file = await onnxProjectDataFolder.CreateFileAsync($"{newModelId.ToString()}.onnx", CreationCollisionOption.ReplaceExisting);

            switch (customVisionProjectType)
            {
            case CustomVisionProjectType.Classification:
                success = await DownloadFileAsync(exportProject.DownloadUri, file);

                break;

            case CustomVisionProjectType.ObjectDetection:
                success = await UnzipModelFileAsync(exportProject.DownloadUri, file);

                break;
            }
            if (!success)
            {
                await file.DeleteAsync();

                return(false);
            }

            string[] classLabels = this.TagsInCurrentGroup?.Select(x => x.Name)?.ToArray() ?? new string[] { };
            newExportedCustomVisionModel = new CustomVisionModelData
            {
                Id          = newModelId,
                Name        = CurrentProject.Name,
                ClassLabels = classLabels,
                ExportDate  = DateTime.UtcNow,
                FileName    = file.Name,
                FilePath    = file.Path
            };

            List <CustomVisionModelData> customVisionModelList = await CustomVisionDataLoader.GetCustomVisionModelDataAsync(customVisionProjectType);

            CustomVisionModelData customVisionModelWithSameName = customVisionModelList.FirstOrDefault(x => string.Equals(x.Name, CurrentProject.Name));

            if (customVisionModelWithSameName != null)
            {
                string titleMessage = $"There is already a “{CurrentProject.Name}” model in this device. Select “Replace” if you would like to replace it, or “Keep Both” if you would like to keep both.";
                await Util.ConfirmActionAndExecute(titleMessage,
                                                   async() =>
                {
                    // if user select Yes, we replace the model with the same name
                    bool modelEntryRemovedFromFile = customVisionModelList.Remove(customVisionModelWithSameName);
                    StorageFile modelFileToRemove  = await onnxProjectDataFolder.GetFileAsync(customVisionModelWithSameName.FileName);
                    if (modelEntryRemovedFromFile && modelFileToRemove != null)
                    {
                        await modelFileToRemove.DeleteAsync();
                    }
                    await SaveCustomVisionModelAsync(customVisionModelList, newExportedCustomVisionModel, customVisionProjectType);

                    // re-display flyout window
                    FlyoutBase.ShowAttachedFlyout(this.exportButton);
                    ShowShareStatus(customVisionProjectType);
                },
                                                   cancelAction : async() =>
                {
                    int maxNumberOfModelWithSameName = customVisionModelList
                                                       .Where(x => x.Name != null && x.Name.StartsWith(newExportedCustomVisionModel.Name, StringComparison.OrdinalIgnoreCase))
                                                       .Select(x =>
                    {
                        string modelNumberInString = x.Name.Split('_').LastOrDefault();
                        int.TryParse(modelNumberInString, out int number);
                        return(number);
                    })
                                                       .Max();

                    // if user select Cancel we just save the new model with the same name
                    newExportedCustomVisionModel.Name = $"{newExportedCustomVisionModel.Name}_{maxNumberOfModelWithSameName + 1}";
                    await SaveCustomVisionModelAsync(customVisionModelList, newExportedCustomVisionModel, customVisionProjectType);

                    // re-display flyout window
                    FlyoutBase.ShowAttachedFlyout(this.exportButton);
                    ShowShareStatus(customVisionProjectType);
                },