Optimized padim #1473
Open
Optimized padim #1473
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
…tions - Added support for MPS backend for running on Apple devices. - Added file type selection support for .pt, .npy, and .pkl formats. - Updated GUI to include: - Enable optimization flag. - Benchmark mode. - File type selection.
…odels into optimized_padim
converted rest of the pipeline to torch
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
added enable_optimization and compare_optimization flags to use optimized code and to compare the results of the optimized code with the original code.
added embedding_concat_optimized that uses PyTorch and performs checkerboard pattern downscaling and concatenation.
added embedding_concat_numpy that uses NumPy and performs checkerboard pattern downscaling and concatenation.
added postprocess_optimized
added infer_optimized that works through vectorization
added infer_init_run to run padim with dummy data to ensure hot inference later
On RTX 4060 Laptop GPU, Optimized code infers default ./bottle_000.png in average 10.25 ms while original code takes 203.0 ms