Example #1
0
        public static async Task RunAi(string outpath, AI ai, bool stepByStep = false)
        {
            Program.mainForm.SetStatus("Downloading models...");
            await ModelDownloader.DownloadModelFiles(ai.pkgDir, current.model);

            if (canceled)
            {
                return;
            }

            currentlyUsingAutoEnc = Utils.CanUseAutoEnc(stepByStep, current);

            IOUtils.CreateDir(outpath);

            List <Task> tasks = new List <Task>();

            if (ai.aiName == Networks.rifeCuda.aiName)
            {
                tasks.Add(AiProcess.RunRifeCuda(current.framesFolder, current.interpFactor, current.model));
            }

            if (ai.aiName == Networks.rifeNcnn.aiName)
            {
                tasks.Add(AiProcess.RunRifeNcnn(current.framesFolder, outpath, (int)current.interpFactor, current.model));
            }

            if (ai.aiName == Networks.flavrCuda.aiName)
            {
                tasks.Add(AiProcess.RunFlavrCuda(current.framesFolder, current.interpFactor, current.model));
            }

            if (ai.aiName == Networks.dainNcnn.aiName)
            {
                tasks.Add(AiProcess.RunDainNcnn(current.framesFolder, outpath, current.interpFactor, current.model, Config.GetInt("dainNcnnTilesize", 512)));
            }

            if (currentlyUsingAutoEnc)
            {
                Logger.Log($"{Logger.GetLastLine()} (Using Auto-Encode)", true);
                tasks.Add(AutoEncode.MainLoop(outpath));
            }

            Program.mainForm.SetStatus("Running AI...");
            await Task.WhenAll(tasks);
        }
Example #2
0
        public async Task CheckModelCacheSize()
        {
            long modelFoldersBytes = 0;

            foreach (string modelFolder in ModelDownloader.GetAllModelFolders())
            {
                modelFoldersBytes += IOUtils.GetDirSize(modelFolder, true);
            }

            if (modelFoldersBytes > 1024 * 1024)
            {
                clearModelCacheBtn.Text = $"Clear Model Cache ({FormatUtils.Bytes(modelFoldersBytes)})";
            }
            else
            {
                clearModelCacheBtn.Enabled = false;
            }
        }
Example #3
0
        public static async Task DownloadForAi(AI ai)
        {
            ModelCollection modelCollection = AiModels.GetModels(ai);

            for (int i = 0; i < modelCollection.models.Count; i++)
            {
                if (canceled)
                {
                    return;
                }

                ModelCollection.ModelInfo modelInfo = modelCollection.models[i];
                form.SetStatus($"Downloading files for {modelInfo.ai.aiName.Replace("_", "-")}...");
                await ModelDownloader.DownloadModelFiles(ai, modelInfo.dir, false);

                taskCounter++;
                UpdateProgressBar();
            }
        }
Example #4
0
    void Start()
    {
        //Directory.CreateDirectory(Application.streamingAssetsPath + "/Test");
        modelManager    = ModelManager.Instance;
        rotationManager = RotationManager.Instance;
        positionManager = PositionManager.Instance;
        scaleManager    = ScaleManager.Instance;

        modelManager.SetModels(ModelDownloader.DownloadModels(inputDirectoryPath, startPoint, parentOfmodels, defaultMaterial));
        if (clearOutputFolder)
        {
            ScreenshotUploader.ClearFolder(outputDirectoryPath);
        }
        ModelObject modelObj = modelManager.GetCurrentModelObject();

        rotationManager.SetCurrentModel(modelObj.gameObject);
        positionManager.SetCurrentModel(modelObj.gameObject);
        scaleManager.SetCurrentModel(modelObj.gameObject);
    }
Example #5
0
 private void clearModelCacheBtn_Click(object sender, EventArgs e)
 {
     ModelDownloader.DeleteAllModels();
     clearModelCacheBtn.Text = "Clear Model Cache";
     CheckModelCacheSize();
 }
Example #6
0
        public static async Task RunAi(string outpath, AI ai, bool stepByStep = false)
        {
            if (canceled)
            {
                return;
            }

            await Task.Run(async() => { await Dedupe.CreateDupesFile(current.framesFolder, currentInputFrameCount, current.framesExt); });

            await Task.Run(async() => { await FrameRename.Rename(); });

            await Task.Run(async() => { await FrameOrder.CreateFrameOrderFile(current.framesFolder, Config.GetBool(Config.Key.enableLoop), current.interpFactor); });

            Program.mainForm.SetStatus("Downloading models...");
            await ModelDownloader.DownloadModelFiles(ai, current.model.dir);

            if (canceled)
            {
                return;
            }

            currentlyUsingAutoEnc = Utils.CanUseAutoEnc(stepByStep, current);

            IoUtils.CreateDir(outpath);

            List <Task> tasks = new List <Task>();

            if (ai.aiName == Implementations.rifeCuda.aiName)
            {
                tasks.Add(AiProcess.RunRifeCuda(current.framesFolder, current.interpFactor, current.model.dir));
            }

            if (ai.aiName == Implementations.rifeNcnn.aiName)
            {
                tasks.Add(AiProcess.RunRifeNcnn(current.framesFolder, outpath, current.interpFactor, current.model.dir));
            }

            if (ai.aiName == Implementations.flavrCuda.aiName)
            {
                tasks.Add(AiProcess.RunFlavrCuda(current.framesFolder, current.interpFactor, current.model.dir));
            }

            if (ai.aiName == Implementations.dainNcnn.aiName)
            {
                tasks.Add(AiProcess.RunDainNcnn(current.framesFolder, outpath, current.interpFactor, current.model.dir, Config.GetInt(Config.Key.dainNcnnTilesize, 512)));
            }

            if (ai.aiName == Implementations.xvfiCuda.aiName)
            {
                tasks.Add(AiProcess.RunXvfiCuda(current.framesFolder, current.interpFactor, current.model.dir));
            }

            if (currentlyUsingAutoEnc)
            {
                Logger.Log($"{Logger.GetLastLine()} (Using Auto-Encode)", true);
                tasks.Add(AutoEncode.MainLoop(outpath));
            }

            Program.mainForm.SetStatus("Running AI...");
            await Task.WhenAll(tasks);
        }
        private static void DownloadDefinitions(Options options)
        {
            var model = new ModelDownloader(options.OctopusUrl, options.OctopusApiKey).DownloadModel();

            new YamlSystemModelRepository().Save(model, options.DefinitionsDir);
        }
 public void SetUp()
 {
     _repository = new FakeOctopusRepository();
     _downloader = new ModelDownloader(_repository, new NullLoggerFactory());
     _uploader   = new ModelUploader(_repository, new NullLoggerFactory());
 }
Example #9
0
 public void SetUp()
 {
     _repository = new FakeOctopusRepository();
     _downloader = new ModelDownloader(_repository);
     _uploader   = new ModelUploader(_repository);
 }