Adding OpenVINO Exporter - #2

Merged

DimaBir merged 94 commits into

Oct 6, 2023

Commits on Oct 2, 2023

Added ONNX exporter class to export model to ONNX format
DimaBir
committed
ab54824
Merge remote-tracking branch 'origin/dev' into dev
DimaBir
committed
c5c5414
Fixed import typo
DimaBir
committed
8ed72b9
Fixed docstring
DimaBir
committed
6451458
Fixed docstring
DimaBir
committed
2dc2560
Merge remote-tracking branch 'origin/dev' into dev
DimaBir
committed
9019460
Fixed typo
DimaBir
committed
758f964
Fixed typo
DimaBir
committed
e186b66
Fixed typo
DimaBir
committed
1a6cf6e
Print ONNX model
DimaBir
committed
47d6481
trying set to train to print BN
DimaBir
committed
74577ea
Removed Conv + BN fusion in exporting PyTorch to ONNX
DimaBir
committed
d00d16f
Removed Conv + BN fusion in exporting PyTorch to ONNX
DimaBir
committed
b07c628
Add ONNX Inference
DimaBir
committed
8e9d013
Updated dockerfile include packages
DimaBir
committed
f04011d
Fixed ONNX Inference
DimaBir
committed
7e04d3b
Fixed ONNX input
DimaBir
committed
85aebba
Fixed ONNX input
DimaBir
committed
7bb680d
Fixed ONNX input
DimaBir
committed
3dc11ca
Fixed ONNX input
DimaBir
committed
6e021cd
Fixed ONNX input
DimaBir
committed
5ac5b47
Fixed ONNX input
DimaBir
committed
69f3c9b
Fixed ONNX input
DimaBir
committed
d0f8936
Fixed ONNX input
DimaBir
committed
d665ae4
Added abstract benchmark class
DimaBir
committed
f0049d8
Fixed ONNXBenchmark param
DimaBir
committed
647d811
Fixed ONNXBenchmark param
DimaBir
committed
125d825
Fixed ONNXBenchmark param
DimaBir
committed
914da47
Fixed ONNXBenchmark param
DimaBir
committed
f3c162d
Applied black formatting
DimaBir
committed
ff84523
Added image cat3
DimaBir
committed
057b899
Enabling optimization for ONNX exporter, using GPU
DimaBir
committed
dd81da6
Enabling optimization for ONNX exporter, using GPU
DimaBir
committed
284671b
Enabling optimization for ONNX exporter, using GPU
DimaBir
committed
e8427c7
Fixed ONNX benchmark error
DimaBir
committed
5acb999
Fixed ONNX benchmark error
DimaBir
committed
e9b08e9
Fixed ONNX benchmark error
DimaBir
committed
3bb4f70
Added requirement, reformatted code
DimaBir
committed
dc574bd
Fixed typo in default image extension
DimaBir
committed
f7a8779
Update inference images
DimaBir
committed
eda33aa
Update inference images
DimaBir
committed
df0bf5f
Updated README.md
DimaBir
committed
e1fefc2
Updated README.md
DimaBir
committed
759e92f
Merge branch 'main' into dev
DimaBir
authored
88c2901

Commits on Oct 5, 2023

Added OV (OpenVINO) exporter model and benchmark
DimaBir
committed
2e5af4c
Update arg parse to handle runtime modes (ONNX, OV, CUDA)
DimaBir
committed
15ba744
Updated dockerfile and requirements for torch-tensorrt
DimaBir
committed
8a1eb3e
Fixed OpenVINO exports
DimaBir
committed
720d793
Fixed OpenVINO exports
DimaBir
committed
75f65cb
Fixed OpenVINO exports
DimaBir
committed
e54f8ab
Fixed OpenVINO typo
DimaBir
committed
0299d79
Fixed OpenVINO typo
DimaBir
committed
8733657
Fixed OpenVINO typo
DimaBir
committed
ddd45a7

Commits on Oct 6, 2023

refactored OV code
DimaBir
committed
d4df36d
Fixed typo in the command
DimaBir
committed
2a433d0
Updated imports
DimaBir
committed
3a16603
Fixed OpenVINO inference by adjusting to the newer API's CompiledModel invocation method.
DimaBir
committed
8bfbcba
Fix
DimaBir
committed
3d2cdb5
Fix and reformat
DimaBir
committed
25b98fa
Fix and reformat
DimaBir
committed
1a3a6a9
Make prediction with OVModel
DimaBir
committed
3d1cb87
Update Dockerfile
DimaBir
committed
605b195
Update Dockerfile
DimaBir
committed
858859b
Update Dockerfile
DimaBir
committed
17d7114
Update Dockerfile
DimaBir
committed
df31b4e
Update Dockerfile
DimaBir
committed
8649bc1
Update Dockerfile
DimaBir
committed
b3262f9
Update Dockerfile
DimaBir
committed
24766d0
Update Dockerfile
DimaBir
committed
1ea7673
Fix: Use correct attribute to retrieve input name in OpenVINO's CompiledModel
DimaBir
committed
bcc1e01
Added: ploting graph and running all
DimaBir
committed
1bc6d5e
Added: ploting graph and running all
DimaBir
committed
e263f04
FIX: OVExport
DimaBir
committed
106e9b8
FIX: OVExport
DimaBir
committed
f6ce7e7
FIX: OVExport
DimaBir
committed
7ece256
Optimized the `make_prediction` function for clarity and stability
DimaBir
committed
e7d5b2a
Revert
DimaBir
committed
5ffd98a
Revert
DimaBir
committed
a7857b4
Revert
DimaBir
committed
99e8a51
Revert
DimaBir
committed
7e17ef9
Added run_all_benchmark
DimaBir
committed
522fe55
Added run_all_benchmark
DimaBir
committed
0cf90b8
Added run_all_benchmark
DimaBir
committed
c0b21d2
Updated run_all_benchmark
DimaBir
committed
d99b076
Updated run_all_benchmark
DimaBir
committed
621c0e7
Updated run_all_benchmark
DimaBir
committed
01caadd
Updated run_all_benchmark
DimaBir
committed
386ecf5
Enhanced benchmark visualization with value labels on bars.
DimaBir
committed
04dd216
Enhanced benchmark visualization with value labels on bars.
DimaBir
committed
c2e6251
Changed README.md
DimaBir
committed
1c1d000
Updated README.md
DimaBir
committed
8f66fd0
Updated README.md
DimaBir
committed
3960263
Updated README.md
DimaBir
committed
5aa4fe0
Merge branch 'master' into dev
DimaBir
authored
3ea712b