RunningLeon
72923e7844
update mmpose ( #1213 )
2022-10-18 10:24:19 +08:00
RunningLeon
6e9c2527b4
merge master from c35099ef0
2022-10-13 11:12:07 +08:00
Qingren
fd21b98efa
[Docs] add the support information about MMPose Hourglass ( #1118 )
...
* [Docs] add the support information about MMPose Hourglass
* * add Hourglass support config
* update benchmark doc
* modify Hourglass AP under pytorch backend
* add regression test for Hourglass
* *update 03-benchmark/benchmark.md
* * modify mmpose.yml to ignore ncnn int8 test
* add trt_fp16 test results
* * add openvino test result
* * modify supported_models.md
2022-10-10 16:38:54 +08:00
RunningLeon
41fcc2242f
log cmd info for regresstion test ( #1146 )
...
* add log to regresstion test
* update for win
* update mmseg reg yml
2022-10-10 16:23:24 +08:00
RunningLeon
51f4e65185
fix reg test for 2.0 ( #1116 )
...
* fix reg
fix mmedit mmocr
update
update apcnet of mmseg
catch error when download fails
log error message
* update reg for win
* update yml
2022-10-08 19:14:33 +08:00
RunningLeon
2e459dba6d
support mmpose2.0 for sdk ( #1080 )
...
* add mmpose sdk
* --amend
* remove debug draw code
* fix docstring
* update
2022-10-08 15:49:44 +08:00
RunningLeon
b78121a776
fix conflicts with dev-1.x
2022-09-30 10:31:32 +08:00
hanrui1sensetime
161f27d493
fix mmedit test picture ( #1131 )
2022-09-29 17:13:23 +08:00
AllentDan
23e84f46ff
update mmcls yaml ( #1137 )
2022-09-29 17:09:17 +08:00
hanrui1sensetime
7f70d7fe56
[Enhancement 2.0] mmdeploy for mmyolo ( #1088 )
...
* support for external codebase like mmyolo
* support for external export
* fix missing flake8
* fix comments
* add aenum
* add missing files
* fix condition
* refactor import_codebase
* fix mmyolo support
* fix lint
* add base codebase
* fix a strange clang-format
* fix import_codebase
* fix dependent codebase register
* wrap custom_model
* fix comment
* add ut
2022-09-29 16:40:47 +08:00
AllentDan
4fc8828af8
fix ocr UT
...
fix ci and lint
fix det
fix cuda ci
fix mmdet test
update object detection
fix ut
fix layer norm ut
update ut
lock mmeit version
fix mmocr mmcls ut
add conftest.py
fix ocr ut
fix mmedit ci
install mmedit from source
fix rknn model and prepare_onnx_paddings__tensorrt UT
docstring
fix coreml export
update mmocr config
small test
recovery assert
fix ci
2022-09-29 16:37:36 +08:00
RunningLeon
144fcf7b3a
fix github ci
...
add mim
update circle ci
fix tests
fix ci
2022-09-29 16:26:31 +08:00
q.yao
d282503012
update detyaml ( #1133 )
2022-09-29 12:14:06 +08:00
hanrui1sensetime
d7cbb395da
[Enhancement 2.0] mmdeploy for mmyolo ( #1088 )
...
* support for external codebase like mmyolo
* support for external export
* fix missing flake8
* fix comments
* add aenum
* add missing files
* fix condition
* refactor import_codebase
* fix mmyolo support
* fix lint
* add base codebase
* fix a strange clang-format
* fix import_codebase
* fix dependent codebase register
* wrap custom_model
* fix comment
* add ut
2022-09-28 16:30:29 +08:00
Li Zhang
bf887cc8bc
[Enhancement] Refactor SDK pipeline ( #938 )
...
* unify C API naming
* fix demo and move apis/c/* -> apis/c/mmdeploy/*
* fix lint
* WIP refactor pipeline
* backward compatibility
* working pipeline demo
* add text det-recog demo
* add det-pose demo
* fix build
* fix demo
* add environment interface
* add environment to pass scheduler & model info at runtime
* update demos
* add pipeline API for Python
* fix `FromPyObject`
* fix for opencv-4.2
* environment -> context, improve pipeline
* python model interface
* fix cmake
* fix python & cmake
* context & C++ pipeline API
* minor fix
* improve API
* fix shared libs
* refresh C/Python API
* propagate context
* fix python demo
* fix
* add namespace
* fix namespace
* fix mis-changed strings
* fix
* fix python api
* rename
* clean-up
* fix pose detector
* clean-up
* clean-up
* clean-up
* fix python API build
* fix CI
* fix lint
* fix lint
* fix lint & demo
* install pipeline.hpp
* fix MSVC shared library build
* fix sample
* fix MSVC monolithic build
* minor fix
2022-09-26 16:11:14 +08:00
RunningLeon
40adb329a3
Merge master:ea7706cb into sync_master
2022-09-22 19:49:50 +08:00
q.yao
ea7706cbfd
fix layer norm ( #1015 )
2022-09-21 16:24:47 +08:00
AllentDan
e3859004ba
add mod rewriter for TRT ( #1024 )
2022-09-21 14:47:36 +08:00
hanrui1sensetime
5c87dd9565
[2.0] Support mmedit 2.0 ( #1017 )
...
* mmcv.Config -> mmengine Config
* support mmedit part
* add rewriter for BaseEditModels
* fix visualizer
* mmedit visualization
* remove unused code
* fix realesrgan
* fix trt
* support MultiTestLoop; rewriter fix mmediting bugs; fix ut
* fix uts
* fix mmedit sdk
* fix regression test(part)
* fix torchscript
* part of fix regression test
* fix checkenv.py
* fix test.py for mmedit2.0
* support for mmedit
* fix regression_test
* fix check copyright ci
* fix isort
* fix docformatter
* fix yapf
* fix tests
* fix sdk after 1040
* add a file for ut
* fix docformatter
* fix export info
* fix super_resolution
* fix test.py
* stage configs
* remove unused code
* remove rewriter of multitestloop
* fix yapf
2022-09-20 19:22:55 +08:00
RunningLeon
2fdba2c523
Support mmpose:dev-1.x ( #884 )
...
* modify mmpose rewritings
* test exporting to ort with hrnet
* support mspn
* update tests
* update tests
* support flip_test
* support mmpose 1.x
* update mmpose.yml
* fix adaptive pool
* align with master for adaptive_pool rewriting
* fix pipeline
* add batch_size to test
* resolve comment
2022-09-19 21:40:53 +08:00
AllentDan
97e0d1228f
SDK ocr 2.0 ( #1006 )
...
* add deploy runner
* fix text_det wrapper
* fix recog
* save
* add crnn support
* update with_padding
* add short scale aspect jitter
* update regression test
* torch2ts
* add test data
* resolve comments
2022-09-19 15:08:51 +08:00
q.yao
0caeaf238c
[CustomOps] TensorRT Gather Topk Ops ( #1033 )
...
* add gather topk
* add shape inference and document
* fix faster rcnn
* reshape topk
* fix
2022-09-19 13:48:26 +08:00
hanrui1sensetime
b0b502cba0
[Fix] Sync codebase 2.0 ( #979 )
...
* remove mmcv.Config
* fix mmdet with new mmengine
* fix ut
* fix mmdet ut
* fix sdk cpp pipeline
* fix mmcls
* fix mmcls ut
* adapt for new config
* fix cls models
* fix mmcls sdk
* fix yapf
* fix lint
* fix torchscript
* fix torchscript cpu
* fix check_env
* add test data
* fix export_info
* fix blank.jpg
* fix mmdet torchscript
* fix test.py
* fix sdk forward
* fix yapf
2022-09-15 21:54:57 +08:00
RunningLeon
06028d6a21
Support mmseg:dev-1.x ( #790 )
...
* support pspnet + ort
* add rewriting for adapt_avg_pool
* test pspnet
* resize seg_pred to original image shape
* run with test.py
* keep as original
* fix ut of segmentation
* update var name
* fix export to torchscript
* sync with mmseg:test-1.x branch
* fix ut
* fix regression test for mmseg
* fix mmseg.ops
* update mmseg yml
* fix mmseg2.0 sdk
* fix adaptive pool
* update rewriting and tests
* fix sdk inputs
2022-09-14 20:08:52 +08:00
AllentDan
124635ec5f
Add RKNN support. ( #865 )
...
* save codes
* support resnet and yolov3
* support yolox
* fix lint
* add mmseg support and a doc
* add UT
* update supported model list
* fix ci
* refine docstring
* resolve comments
* remote output_tensor_type
* resolve comments
* update readme
2022-09-06 11:48:39 +08:00
Chen Xin
6b01a2e649
[Feature] Add option to fuse transform. ( #741 )
...
* add collect_impl.cpp to cuda device
* add dummy compute node wich device elena
* add compiler & dynamic library loader
* add code to compile with gen code(elena)
* move folder
* fix lint
* add tracer module
* add license
* update type id
* add fuse kernel registry
* remove compilier & dynamic_library
* update fuse kernel interface
* Add elena-mmdeploy project in 3rd-party
* Fix README.md
* fix cmake file
* Support cuda device and clang format all file
* Add cudaStreamSynchronize for cudafree
* fix cudaStreamSynchronize
* rename to __tracer__
* remove unused code
* update kernel
* update extract elena script
* update gitignore
* fix ci
* Change the crop_size to crop_h and crop_w in arglist
* update Tracer
* remove cond
* avoid allocate memory
* add build.sh for elena
* remove code
* update test
* Support bilinear resize with float input
* Rename elena-mmdeploy to delete
* Introduce public submodule
* use get_ref
* update elena
* update tools
* update tools
* update fuse transform docs
* add fuse transform doc link to get_started
* fix shape in crop
* remove fuse_transform_ == true check
* remove fuse_transform_ member
* remove elena_int.h
* doesn't dump transform_static.json
* update tracer
* update CVFusion to remove compile warning
* remove mmcv version > 1.5.1 dep
* fix tests
* update docs
* add elena use option
* remove submodule of CVFusion
* update doc
* use auto
* use throw_exception(eEntryNotFound);
* update
Co-authored-by: cx <cx@ubuntu20.04>
Co-authored-by: miraclezqc <969226879@qq.com>
2022-09-05 20:29:18 +08:00
RunningLeon
ac3a12026d
fix mmdet ut ( #1001 )
2022-09-05 19:56:19 +08:00
q.yao
9541be9c0b
TensorRT dot product attention ops ( #949 )
...
* add detr support
* fix softmax
* add placeholder
* add implement
* add docs and ut
* update testcase
* update docs
* update docs
2022-09-05 18:25:39 +08:00
Li Zhang
792c27b054
[Feature] Ascend backend ( #747 )
...
* add acl backend
* support dynamic batch size and dynamic image size
* add preliminary ascend backend
* support dtypes other than float
* support dynamic_dims in SDK
* fix dynamic batch size
* better error handling
* remove debug info
* [WIP] dynamic shape support
* fix static shape
* fix dynamic batch size
* add retinanet support
* fix dynamic image size
* fix dynamic image size
* fix dynamic dims
* fix dynamic dims
* simplify config files
* fix yolox support
* fix negative index
* support faster rcnn
* add seg config
* update benchmark
* fix onnx2ascend dynamic shape
* update docstring and benchmark
* add unit test, update documents
* fix wrapper
* fix ut
* fix for vit
* error handling
* context handling & multi-device support
* build with stub libraries
* add ci
* fix lint
* fix lint
* update doc ref
* fix typo
* down with `target_link_directories`
* setup python
* makedir
* fix ci
* fix ci
* remove verbose logs
* fix UBs
* export Error
* fix lint
* update checkenv
Co-authored-by: grimoire <yaoqian@sensetime.com>
2022-09-05 12:08:36 +08:00
RunningLeon
966d737a1b
Fix: skip tests for uninstalled codebases ( #987 )
...
* skip tests if codebase not installed
* skip ort run test
* fix mmseg
2022-09-03 15:07:08 +08:00
RunningLeon
cbedf1cc30
fix regression test ( #958 )
...
* fix reg
* set sdk wrapper device id
* resolve comment
2022-09-02 11:28:12 +08:00
RunningLeon
71e64786b7
Fix mmseg pointrend ( #903 )
...
* support mmseg:pointrend
* update docs
* update docs for torchscript
* resolve comments
2022-09-02 10:55:32 +08:00
AllentDan
a2f82874bb
Support mmocr:dev-1.x ( #904 )
...
* init
* update UT
* fix UT except SAR
* update to latest 2.0
* fix ncnn UT
* export info
2022-09-01 15:11:43 +08:00
hanrui1sensetime
308e28fcb0
[Enhancement] Support Object Detection and Instance Segmentation for ort trt ncnn and openvino in mmdet 2.0 ( #786 )
...
* support cascade (mask) rcnn
* fix docstring
* support SwinTransformer
* move dense_head support to this branch
* fix function names
* fix part of uts of mmdet
* fix for mmdet ut
* fix det model cfg for ut
* fix test_object_detection.py
* fix mmdet object_detection_model.py
* fix mmdet yolov3 ort ut
* fix part of uts
* fix cascade bbox head ut
* fix cascade bbox head ut
* remove useless ssd ncnn test
* fix ncnn wrapper
* fix openvino ut for reppoint head
* fix openvino cascade mask rcnn
* sync codes
* support roll
* remove unused pad
* fix yolox
* fix isort
* fix lint
* fix flake8
* reply for comments and fix failed ut
* fix sdk_export in dump_info
* fix temp hidden xlsx bugs
* fix mmdet regression test
* fix lint
* fix timer
* fix timecount side-effect
* adapt profile.py for mmdet 2.0
* hardcode report.txt for T4 benchmark test: temp version
* fix no-visualizer case
* fix backend_model
* fix android build
* adapt new mmdet 2.0 0825
* fix new 2.0
* fix test_mmdet_structures
* fix test_object_detection
* fix codebase import
* fix ut
* fix all mmdet uts
* fix det
* fix mmdet trt
* fix ncnn onnx optimize
2022-09-01 11:35:57 +08:00
q.yao
eed69b14e2
fix ci ( #964 )
2022-08-31 14:36:15 +08:00
q.yao
13920ec1a2
[Enhancement] Rewriter support pre-import function ( #899 )
...
* support preimport
* update rewriter
* fix batched nms ort
2022-08-30 12:32:43 +08:00
hanrui1sensetime
47d4e6f733
fix ut failed ( #951 )
2022-08-30 11:21:18 +08:00
q.yao
f4decda86e
[Enhancement] Support DETR ( #924 )
...
* add detr support
* fix softmax
* add reg test, update document
2022-08-30 10:39:10 +08:00
q.yao
18321eed51
[Bug] fix setitem with scalar or single element tensor ( #941 )
...
* fix setitem
* add copy symbolic
2022-08-29 18:56:17 +08:00
AllentDan
0111e1f10f
[Feature] add swin for cls ( #911 )
...
* add swin for cls
* add ut and doc
* reduce trt batch size
* add regression test
* resolve comments
* remove useless rewriting logic
2022-08-23 17:14:24 +08:00
RunningLeon
4d8ea40f55
Sync v0.7.0 to dev-1.x ( #907 )
...
* make -install -> make install (#621 )
change `make -install` to `make install`
https://github.com/open-mmlab/mmdeploy/issues/618
* [Fix] fix csharp api detector release result (#620 )
* fix csharp api detector release result
* fix wrong count arg of xxx_release_result in c# api
* [Enhancement] Support two-stage rotated detector TensorRT. (#530 )
* upload
* add fake_multiclass_nms_rotated
* delete unused code
* align with pytorch
* Update delta_midpointoffset_rbbox_coder.py
* add trt rotated roi align
* add index feature in nms
* not good
* fix index
* add ut
* add benchmark
* move to csrc/mmdeploy
* update unit test
Co-authored-by: zytx121 <592267829@qq.com>
* Reduce mmcls version dependency (#635 )
* fix shufflenetv2 with trt (#645 )
* fix shufflenetv2 and pspnet
* fix ci
* remove print
* ' -> " (#654 )
If there is a variable in the string, single quotes will ignored it, while double quotes will bring the variable into the string after parsing
* ' -> " (#655 )
same with https://github.com/open-mmlab/mmdeploy/pull/654
* Support deployment of Segmenter (#587 )
* support segmentor with ncnn
* update regression yml
* replace chunk with split to support ts
* update regression yml
* update docs
* fix segmenter ncnn inference failure brought by #477
* add test
* fix test for ncnn and trt
* fix lint
* export nn.linear to Gemm op in onnx for ncnn
* fix ci
* simplify `Expand` (#617 )
* Fix typo (#625 )
* Add make install in en docs
* Add make install in zh docs
* Fix typo
* Merge and add windows build
Co-authored-by: tripleMu <865626@163.com>
* [Enhancement] Fix ncnn unittest (#626 )
* optmize-csp-darknet
* replace floordiv to torch.div
* update csp_darknet default implement
* fix test
* [Enhancement] TensorRT Anchor generator plugin (#646 )
* custom trt anchor generator
* add ut
* add docstring, update doc
* Add partition doc and sample code (#599 )
* update torch2onnx tool to support onnx partition
* add model partition of yolov3
* add cn doc
* update torch2onnx tool to support onnx partition
* add model partition of yolov3
* add cn doc
* add to index.rst
* resolve comment
* resolve comments
* fix lint
* change caption level in docs
* update docs (#624 )
* Add java apis and demos (#563 )
* add java classifier detector
* add segmentor
* fix lint
* add ImageRestorer java apis and demo
* remove useless count parameter for Segmentor and Restorer, add PoseDetector
* add RotatedDetection java api and demo
* add Ocr java demo and apis
* remove mmrotate ncnn java api and demo
* fix lint
* sync java api folder after rebase to master
* fix include
* remove record
* fix java apis dir path in cmake
* add java demo readme
* fix lint mdformat
* add test javaapi ci
* fix lint
* fix flake8
* fix test javaapi ci
* refactor readme.md
* fix install opencv for ci
* fix install opencv : add permission
* add all codebases and mmcv install
* add torch
* install mmdeploy
* fix image path
* fix picture path
* fix import ncnn
* fix import ncnn
* add submodule of pybind
* fix pybind submodule
* change download to git clone for submodule
* fix ncnn dir
* fix README error
* simplify the github ci
* fix ci
* fix yapf
* add JNI as required
* fix Capitalize
* fix Capitalize
* fix copyright
* ignore .class changed
* add OpenJDK installation docs
* install target of javaapi
* simplify ci
* add jar
* fix ci
* fix ci
* fix test java command
* debugging what failed
* debugging what failed
* debugging what failed
* add java version info
* install openjdk
* add java env var
* fix export
* fix export
* fix export
* fix export
* fix picture path
* fix picture path
* fix file name
* fix file name
* fix README
* remove java_api strategy
* fix python version
* format task name
* move args position
* extract common utils code
* show image class result
* add detector result
* segmentation result format
* add ImageRestorer result
* add PoseDetection java result format
* fix ci
* stage ocr
* add visualize
* move utils
* fix lint
* fix ocr bugs
* fix ci demo
* fix java classpath for ci
* fix popd
* fix ocr demo text garbled
* fix ci
* fix ci
* fix ci
* fix path of utils ci
* update the circleci config file by adding workflows both for linux, windows and linux-gpu (#368 )
* update circleci by adding more workflows
* fix test workflow failure on windows platform
* fix docker exec command for SDK unittests
* Fixed tensorrt plugin not found in Windows (#672 )
* update introduction.png (#674 )
* [Enhancement] Add fuse select assign pass (#589 )
* Add fuse select assign pass
* move code to csrc
* add config flag
* remove bool cast
* fix export sdk info of input shape (#667 )
* Update get_started.md (#675 )
Fix backend model assignment
* Update get_started.md (#676 )
Fix backend model assignment
* [Fix] fix clang build (#677 )
* fix clang build
* fix ndk build
* fix ndk build
* switch to `std::filesystem` for clang-7 and later
* Deploy the Swin Transformer on TensorRT. (#652 )
* resolve conflicts
* update ut and docs
* fix ut
* refine docstring
* add comments and refine UT
* resolve comments
* resolve comments
* update doc
* add roll export
* check backend
* update regression test
* bump version to 0.6.0 (#680 )
* bump vertion to 0.6.0
* update version
* pass img_metas while exporting to onnx (#681 )
* pass img_metas while exporting to onnx
* remove try-catch in tools for beter debugging
* use get
* fix typo
* [Fix] fix ssd ncnn ut (#692 )
* fix ssd ncnn ut
* fix yapf
* fix passing img_metas to pytorch2onnx for mmedit (#700 )
* fix passing img_metas for mmdet3d (#707 )
* [Fix] Fix android build (#698 )
* fix android build
* fix cmake
* fix url link
* fix wrong exit code in pipeline_manager (#715 )
* fix exit
* change to general exit errorcode=1
* fix passing wrong backend type (#719 )
* Rename onnx2ncnn to mmdeploy_onnx2ncnn (#694 )
* improvement(tools/onnx2ncnn.py): rename to mmdeploy_onnx2ncnn
* format(tools/deploy.py): clean code
* fix(init_plugins.py): improve if condition
* fix(CI): update target
* fix(test_onnx2ncnn.py): update desc
* Update init_plugins.py
* [Fix] Fix mmdet ort static shape bug (#687 )
* fix shape
* add device
* fix yapf
* fix rewriter for transforms
* reverse image shape
* fix ut of distance2bbox
* fix rewriter name
* fix c4 for torchscript (#724 )
* [Enhancement] Standardize C API (#634 )
* unify C API naming
* fix demo and move apis/c/* -> apis/c/mmdeploy/*
* fix lint
* fix C# project
* fix Java API
* [Enhancement] Support Slide Vertex TRT (#650 )
* reorgnize mmrotate
* fix
* add hbb2obb
* add ut
* fix rotated nms
* update docs
* update benchmark
* update test
* remove ort regression test, remove comment
* Fix get-started rendering issues in readthedocs (#740 )
* fix mermaid markdown rendering issue in readthedocs
* fix error in C++ example
* fix error in c++ example in zh_cn get_started doc
* [Fix] set default topk for dump info (#702 )
* set default topk for dump info
* remove redundant docstrings
* add ci densenet
* fix classification warnings
* fix mmcls version
* fix logger.warnings
* add version control (#754 )
* fix satrn for ORT (#753 )
* fix satrn for ORT
* move rewrite into pytorch
* Add inference latency test tool (#665 )
* add profile tool
* remove print envs in profile tool
* set cudnn_benchmark to True
* add doc
* update tests
* fix typo
* support test with images from a directory
* update doc
* resolve comments
* [Enhancement] Add CSE ONNX pass (#647 )
* Add fuse select assign pass
* move code to csrc
* add config flag
* Add fuse select assign pass
* Add CSE for ONNX
* remove useless code
* Test robot
Just test robot
* Update README.md
Revert
* [Fix] fix yolox point_generator (#758 )
* fix yolox point_generator
* add a UT
* resolve comments
* fix comment lines
* limit markdown version (#773 )
* [Enhancement] Better index put ONNX export. (#704 )
* Add rewriter for tensor setitem
* add version check
* Upgrade Dockerfile to use TensorRT==8.2.4.2 (#706 )
* Upgrade TensorRT to 8.2.4.2
* upgrade pytorch&mmcv in CPU Dockerfile
* Delete redundant port example in Docker
* change 160x160-608x608 to 64x64-608x608 for yolov3
* [Fix] reduce log verbosity & improve error reporting (#755 )
* reduce log verbosity & improve error reporting
* improve error reporting
* [Enhancement] Support latest ppl.nn & ppl.cv (#564 )
* support latest ppl.nn
* fix pplnn for model convertor
* fix lint
* update memory policy
* import algo from buffer
* update ppl.cv
* use `ppl.cv==0.7.0`
* document supported ppl.nn version
* skip pplnn dependency when building shared libs
* [Fix][P0] Fix for torch1.12 (#751 )
* fix for torch1.12
* add comment
* fix check env (#785 )
* [Fix] fix cascade mask rcnn (#787 )
* fix cascade mask rcnn
* fix lint
* add regression
* [Feature] Support RoITransRoIHead (#713 )
* [Feature] Support RoITransRoIHead
* Add docs
* Add mmrotate models regression test
* Add a draft for test code
* change the argument name
* fix test code
* fix minor change for not class agnostic case
* fix sample for test code
* fix sample for test code
* Add mmrotate in requirements
* Revert "Add mmrotate in requirements"
This reverts commit 043490075e
.
* [Fix] fix triu (#792 )
* fix triu
* triu -> triu_default
* [Enhancement] Install Optimizer by setuptools (#690 )
* Add fuse select assign pass
* move code to csrc
* add config flag
* Add fuse select assign pass
* Add CSE for ONNX
* remove useless code
* Install optimizer by setup tools
* fix comment
* [Feature] support MMRotate model with le135 (#788 )
* support MMRotate model with le135
* cse before fuse select assign
* remove unused import
* [Fix] Support macOS build (#762 )
* fix macOS build
* fix missing
* add option to build & install examples (#822 )
* [Fix] Fix setup on non-linux-x64 (#811 )
* fix setup
* replace long to int64_t
* [Feature] support build single sdk library (#806 )
* build single lib for c api
* update csharp doc & project
* update test build
* fix test build
* fix
* update document for building android sdk (#817 )
Co-authored-by: dwSun <dwsunny@icloud.com>
* [Enhancement] support kwargs in SDK python bindings (#794 )
* support-kwargs
* make '__call__' as single image inference and add 'batch' API to deal with batch images inference
* fix linting error and typo
* fix lint
* improvement(sdk): add sdk code coverage (#808 )
* feat(doc): add CI
* CI(sdk): add sdk coverage
* style(test): code format
* fix(CI): update coverage.info path
* improvement(CI): use internal image
* improvement(CI): push coverage info once
* [Feature] Add C++ API for SDK (#831 )
* add C++ API
* unify result type & add examples
* minor fix
* install cxx API headers
* fix Mat, add more examples
* fix monolithic build & fix lint
* install examples correctly
* fix lint
* feat(tools/deploy.py): support snpe (#789 )
* fix(tools/deploy.py): support snpe
* improvement(backend/snpe): review advices
* docs(backend/snpe): update build
* docs(backend/snpe): server support specify port
* docs(backend/snpe): update path
* fix(backend/snpe): time counter missing argument
* docs(backend/snpe): add missing argument
* docs(backend/snpe): update download and using
* improvement(snpe_net.cpp): load model with modeldata
* Support setup on environment with no PyTorch (#843 )
* support test with multi batch (#829 )
* support test with multi batch
* resolve comment
* import algorithm from buffer (#793 )
* [Enhancement] build sdk python api in standard-alone manner (#810 )
* build sdk python api in standard-alone manner
* enable MMDEPLOY_BUILD_SDK_MONOLITHIC and MMDEPLOY_BUILD_EXAMPLES in prebuild config
* link mmdeploy to python target when monolithic option is on
* checkin README to describe precompiled package build procedure
* use packaging.version.parse(python_version) instead of list(python_version)
* fix according to review results
* rebase master
* rollback cmake.in and apis/python/CMakeLists.txt
* reorganize files in install/example
* let cmake detect visual studio instead of specifying 2019
* rename whl name of precompiled package
* fix according to review results
* Fix SDK backend (#844 )
* fix mmpose python api (#852 )
* add prebuild package usage docs on windows (#816 )
* add prebuild package usage docs on windows
* fix lint
* update
* try fix lint
* add en docs
* update
* update
* udpate faq
* fix typo (#862 )
* [Enhancement] Improve get_started documents and bump version to 0.7.0 (#813 )
* simplify commands in get_started
* add installation commands for Windows
* fix typo
* limit markdown and sphinx_markdown_tables version
* adopt html <details open> tag
* bump mmdeploy version
* bump mmdeploy version
* update get_started
* update get_started
* use python3.8 instead of python3.7
* remove duplicate section
* resolve issue #856
* update according to review results
* add reference to prebuilt_package_windows.md
* fix error when build sdk demos
* fix mmcls
Co-authored-by: Ryan_Huang <44900829+DrRyanHuang@users.noreply.github.com>
Co-authored-by: Chen Xin <xinchen.tju@gmail.com>
Co-authored-by: q.yao <yaoqian@sensetime.com>
Co-authored-by: zytx121 <592267829@qq.com>
Co-authored-by: Li Zhang <lzhang329@gmail.com>
Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: tripleMu <865626@163.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: Bryan Glen Suello <11388006+bgsuello@users.noreply.github.com>
Co-authored-by: zambranohally <63218980+zambranohally@users.noreply.github.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: tpoisonooo <khj.application@aliyun.com>
Co-authored-by: Hakjin Lee <nijkah@gmail.com>
Co-authored-by: 孙德伟 <5899962+dwSun@users.noreply.github.com>
Co-authored-by: dwSun <dwsunny@icloud.com>
Co-authored-by: Chen Xin <irexyc@gmail.com>
2022-08-19 09:30:13 +08:00
AllentDan
cad0092801
[Fix] fix device error in dump-info ( #912 )
...
* fix device error in dump-info
* fix UT
2022-08-17 18:01:26 +08:00
RunningLeon
670a504502
fix adaptive_avg_pool exporting to onnx ( #857 )
...
* fix adaptive_avg_pool exporting to onnx
* remove debug codes
* fix ci
* resolve comment
2022-08-12 09:38:46 +08:00
VVsssssk
f957284d54
[Feature]support pointpillar nus version ( #391 )
...
* support pointpillar nus version
* support pointpillar nus version
* add regression test config for mmdet3d
* fix exit with no error code
* fix cfg
* fix worksize
* fix worksize
* fix cfg
* support nus pp
* fix yaml
* fix yaml
* fix yaml
* add ut
* fix ut
Co-authored-by: RunningLeon <mnsheng@yeah.net>
2022-08-05 16:37:20 +08:00
tpoisonooo
f54da4e73f
improvement(sdk): add sdk code coverage ( #808 )
...
* feat(doc): add CI
* CI(sdk): add sdk coverage
* style(test): code format
* fix(CI): update coverage.info path
* improvement(CI): use internal image
* improvement(CI): push coverage info once
2022-07-29 13:57:54 +08:00
Chen Xin
b6b22a1b6f
[Feature] support build single sdk library ( #806 )
...
* build single lib for c api
* update csharp doc & project
* update test build
* fix test build
* fix
2022-07-28 14:07:48 +08:00
Li Zhang
bfd46c1c68
[Fix] Support macOS build ( #762 )
...
* fix macOS build
* fix missing
2022-07-26 21:54:49 +08:00
q.yao
0e1a3aa8c7
[Feature] support MMRotate model with le135 ( #788 )
...
* support MMRotate model with le135
* cse before fuse select assign
* remove unused import
2022-07-25 16:01:05 +08:00
q.yao
5b31d7a60d
[Enhancement] Install Optimizer by setuptools ( #690 )
...
* Add fuse select assign pass
* move code to csrc
* add config flag
* Add fuse select assign pass
* Add CSE for ONNX
* remove useless code
* Install optimizer by setup tools
* fix comment
2022-07-25 13:04:27 +08:00
AllentDan
36c35b6e88
[Fix] fix triu ( #792 )
...
* fix triu
* triu -> triu_default
2022-07-22 16:31:00 +08:00
Hakjin Lee
36b3ca4461
[Feature] Support RoITransRoIHead ( #713 )
...
* [Feature] Support RoITransRoIHead
* Add docs
* Add mmrotate models regression test
* Add a draft for test code
* change the argument name
* fix test code
* fix minor change for not class agnostic case
* fix sample for test code
* fix sample for test code
* Add mmrotate in requirements
* Revert "Add mmrotate in requirements"
This reverts commit 043490075e
.
2022-07-22 15:19:37 +08:00
hanrui1sensetime
4b79124a97
[Fix] fix cascade mask rcnn ( #787 )
...
* fix cascade mask rcnn
* fix lint
* add regression
2022-07-22 12:09:12 +08:00
Hakjin Lee
58b256a13b
Upgrade Dockerfile to use TensorRT==8.2.4.2 ( #706 )
...
* Upgrade TensorRT to 8.2.4.2
* upgrade pytorch&mmcv in CPU Dockerfile
* Delete redundant port example in Docker
* change 160x160-608x608 to 64x64-608x608 for yolov3
2022-07-20 20:36:22 +08:00
q.yao
0310c168dc
[Enhancement] Better index put ONNX export. ( #704 )
...
* Add rewriter for tensor setitem
* add version check
2022-07-20 19:32:46 +08:00
AllentDan
ebba5ff6c9
[Fix] fix yolox point_generator ( #758 )
...
* fix yolox point_generator
* add a UT
* resolve comments
* fix comment lines
2022-07-20 14:18:48 +08:00
q.yao
4842823fd0
[Enhancement] Add CSE ONNX pass ( #647 )
...
* Add fuse select assign pass
* move code to csrc
* add config flag
* Add fuse select assign pass
* Add CSE for ONNX
* remove useless code
2022-07-19 14:46:14 +08:00
RunningLeon
252cabbbc5
Add inference latency test tool ( #665 )
...
* add profile tool
* remove print envs in profile tool
* set cudnn_benchmark to True
* add doc
* update tests
* fix typo
* support test with images from a directory
* update doc
* resolve comments
2022-07-19 14:28:22 +08:00
AllentDan
394fb55809
fix satrn for ORT ( #753 )
...
* fix satrn for ORT
* move rewrite into pytorch
2022-07-19 09:47:42 +08:00
hanrui1sensetime
d216af885d
[Fix] set default topk for dump info ( #702 )
...
* set default topk for dump info
* remove redundant docstrings
* add ci densenet
* fix classification warnings
* fix mmcls version
* fix logger.warnings
2022-07-14 17:32:22 +08:00
q.yao
dace58e844
[Enhancement] Support Slide Vertex TRT ( #650 )
...
* reorgnize mmrotate
* fix
* add hbb2obb
* add ut
* fix rotated nms
* update docs
* update benchmark
* update test
* remove ort regression test, remove comment
2022-07-13 16:09:09 +08:00
Li Zhang
14b2bfd524
[Enhancement] Standardize C API ( #634 )
...
* unify C API naming
* fix demo and move apis/c/* -> apis/c/mmdeploy/*
* fix lint
* fix C# project
* fix Java API
2022-07-12 14:04:33 +08:00
grimoire
27a856637c
fix mmcls ut
2022-07-11 19:10:04 +08:00
hanrui1sensetime
112b11469c
[Fix] Fix mmdet ort static shape bug ( #687 )
...
* fix shape
* add device
* fix yapf
* fix rewriter for transforms
* reverse image shape
* fix ut of distance2bbox
* fix rewriter name
2022-07-09 07:20:15 +08:00
hanrui1sensetime
affc5ae3ff
[Fix] fix ssd ncnn ut ( #692 )
...
* fix ssd ncnn ut
* fix yapf
2022-07-04 15:17:30 +08:00
RunningLeon
17a7d60c7d
pass img_metas while exporting to onnx ( #681 )
...
* pass img_metas while exporting to onnx
* remove try-catch in tools for beter debugging
* use get
* fix typo
2022-06-30 17:33:24 +08:00
AllentDan
efd39953d7
Deploy the Swin Transformer on TensorRT. ( #652 )
...
* resolve conflicts
* update ut and docs
* fix ut
* refine docstring
* add comments and refine UT
* resolve comments
* resolve comments
* update doc
* add roll export
* check backend
* update regression test
2022-06-30 10:41:24 +08:00
Li Zhang
8f1508e098
[Fix] fix clang build ( #677 )
...
* fix clang build
* fix ndk build
* fix ndk build
* switch to `std::filesystem` for clang-7 and later
2022-06-29 22:01:51 +08:00
q.yao
58584888ee
[Enhancement] Add fuse select assign pass ( #589 )
...
* Add fuse select assign pass
* move code to csrc
* add config flag
* remove bool cast
2022-06-29 14:43:01 +08:00
grimoire
8c37f79f06
rename method
2022-06-29 14:14:03 +08:00
lvhan028
30d62e97a1
update the circleci config file by adding workflows both for linux, windows and linux-gpu ( #368 )
...
* update circleci by adding more workflows
* fix test workflow failure on windows platform
* fix docker exec command for SDK unittests
2022-06-29 12:05:56 +08:00
RunningLeon
5fce1e8f8d
update docs ( #624 )
2022-06-29 10:15:48 +08:00
q.yao
dc5f9c3746
[Enhancement] TensorRT Anchor generator plugin ( #646 )
...
* custom trt anchor generator
* add ut
* add docstring, update doc
2022-06-28 16:11:24 +08:00
q.yao
4d9e20960d
[Enhancement] Fix ncnn unittest ( #626 )
...
* optmize-csp-darknet
* replace floordiv to torch.div
* update csp_darknet default implement
* fix test
2022-06-28 09:41:44 +08:00
RunningLeon
da8c02e2b4
Support deployment of Segmenter ( #587 )
...
* support segmentor with ncnn
* update regression yml
* replace chunk with split to support ts
* update regression yml
* update docs
* fix segmenter ncnn inference failure brought by #477
* add test
* fix test for ncnn and trt
* fix lint
* export nn.linear to Gemm op in onnx for ncnn
* fix ci
2022-06-27 17:52:05 +08:00
q.yao
0cac5154a6
[Enhancement] Support two-stage rotated detector TensorRT. ( #530 )
...
* upload
* add fake_multiclass_nms_rotated
* delete unused code
* align with pytorch
* Update delta_midpointoffset_rbbox_coder.py
* add trt rotated roi align
* add index feature in nms
* not good
* fix index
* add ut
* add benchmark
* move to csrc/mmdeploy
* update unit test
Co-authored-by: zytx121 <592267829@qq.com>
2022-06-23 10:35:10 +08:00
grimoire
41a97746cf
finish mmcls ut
2022-06-22 22:32:14 +08:00
tpoisonooo
a228f95335
Improve ncnn patch embed ( #592 )
...
* improvement(PatchEmbed): use reshape instead of flatten
* improvement(init_plugins.py): get onnx2ncnn path
* fix(CI): build error
* fix(CI): revert get onnx2ncnn path
2022-06-15 19:16:51 +08:00
Li Zhang
28beb9a481
[Enhancement] Adjust `csrc` structure ( #594 )
...
* adjust csrc structure
* adjust csrc includes
* fix lint
* fix lint
* fix MSVC build
* remove useless code
2022-06-14 13:46:01 +08:00
hanrui1sensetime
c99379cc76
[Fix] Fix ncnn float bugs for segmentor sdk ( #572 )
...
* fix ncnn float bugs for segmentor sdk
* fix segment for all cases
* fix ut
Co-authored-by: RunningLeon <mnsheng@yeah.net>
2022-06-08 22:30:23 +08:00
Chen Xin
0cb2f67b32
Fix mmpose regression test failed on mmpose v0.26.0 ( #576 )
...
* fix mmpose regression test
* try to fix lint
* fix lint
* remove old config
Co-authored-by: chenxin2 <chenxin2@sensetime.com>
2022-06-08 20:16:23 +08:00
lvhan028
594c835b6b
merge master and fix the conflicts
2022-06-07 19:04:29 +08:00
q.yao
74243dc98b
[Enhancement] Add ONNX passes support ( #390 )
...
* add merge_shape_concate
* add some peephole optimization
* bug fixing
* fix for torch1.9
* add flatten cls head
* add subgraph matcher with attribute
* add ut
* fix lint
* remove onnx2ncnn
* add opset version
* axis name
* fix peephole
* fix symbol compare
* add docs
2022-06-06 21:30:31 +08:00
lzhangzz
567ec62296
[Fix] Fix recent build problems ( #544 )
...
* simplify deps management
* skip visibility flags for MSVC
* simplify cuda deps
* naming
* workaround for cmake<3.17
* add spdlog dependency
* move the enablement of CUDA to top level CMakeLists.txt
* fix MSVC build
* fix lint
* fix build for backend ops only
* remove comment
* allow to use apis/python as a standalone project
* remove redundant cmake code
* control shared or static lib using `MMDEPLOY_SHARED_LIBS` instead of `BUILD_SHARED_LIBS`
* fix MSVC build
* update docs
2022-06-06 11:19:34 +08:00
RunningLeon
51f630b22c
Fix test ncnn failure in regression test ( #551 )
...
* fix ncnn test in regression test
* update doc
* fix docstring
2022-06-06 10:00:02 +08:00
tpoisonooo
cd336eada1
improvement(ViT): use Crop to subtitude Gather ( #477 )
...
* improvement(ViT): use Crop to subtitude Gather
* fix(CI): code format
* fix(pytorch/ops/linear.py): bias maybe None
* fix(test/test_pytorch_ops.py): op_type error
* fix(test): pytest error
* fix(test): torch version 1.8
2022-06-02 19:39:15 +08:00
AllentDan
ee878b539b
change resolution and fix regression test for mmocr ( #534 )
2022-06-02 18:29:41 +08:00
q.yao
07fe3e85e2
[RegressionTest] Add mmrotate regression test ( #542 )
...
* Add mmrotate regression test
* update regression test script
2022-06-02 14:28:33 +08:00
lzhangzz
46bfe0ac87
[Feature] New pipeline & executor for SDK ( #497 )
...
* executor prototype
* add split/when_all
* fix GCC build
* WIP let_value
* fix let_value
* WIP ensure_started
* ensure_started & start_detached
* fix let_value + when_all combo on MSVC 142
* fix static thread pool
* generic just, then, let_value, sync_wait
* minor
* generic split and when_all
* fully generic sender adapters
* when_all: workaround for GCC7
* support legacy spdlog
* fix memleak
* bulk
* static detector
* fix bulk & first pipeline
* bulk for static thread pools
* fix on MSVC
* WIP async batch submission
* WIP collation
* async batch
* fix detector
* fix async detector
* fix
* fix
* debug
* fix cuda allocator
* WIP type erased executor
* better type erasure
* simplify C API impl
* Expand & type erase TC
* deduction guide for type erased senders
* fix GCC build
* when_all for arrays of Value senders
* WIP pipeline v2
* WIP pipeline parser
* WIP timed batch operation
* add registry
* experiment
* fix pipeline
* naming
* fix mem-leak
* fix deferred batch operation
* WIP
* WIP configurable scheduler
* WIP configurable scheduler
* add comment
* parse scheduler config
* force link schedulers
* WIP pipeable sender
* WIP CPO
* ADL isolation and dismantle headers
* type erase single thread context
* fix MSVC build
* CPO
* replace decay_t with remove_cvref_t
* structure adjustment
* structure adjustment
* apply CPOs & C API rework
* refine C API
* detector async C API
* adjust detector async C API
* # Conflicts:
# csrc/apis/c/detector.cpp
* fix when_all for type erased senders
* support void return for Then
* async detector
* fix some CPOs
* minor
* WIP rework capture mechanism for type erased types
* minor fix
* fix MSVC build
* move expand.h to execution
* make `Expand` pipeable
* fix type erased
* un-templatize `_TypeErasedOperation`
* re-work C API
* remove async_detector C API
* fix pipeline
* add flatten & unflatten
* fix flatten & unflatten
* add aync OCR demo
* config executor for nodes & better executor API
* working async OCR example
* minor
* dynamic batch via scheduler
* dynamic batch on `Value`
* fix MSVC build
* type erase dynamic batch scheduler
* sender as Python Awaitable
* naming
* naming
* add docs
* minor
* merge tmp branch
* unify C APIs
* fix ocr
* unify APIs
* fix typo
* update async OCR demo
* add v3 API text recognizer
* fix v3 API
* fix lint
* add license info & reformat
* add demo async_ocr_v2
* revert files
* revert files
* resolve link issues
* fix scheduler linkage for shared libs
* fix license header
* add docs for `mmdeploy_executor_split`
* add missing `mmdeploy_executor_transfer_just` and `mmdeploy_executor_execute`
* make `TimedSingleThreadContext` header only
* fix lint
* simplify type-erased sender
2022-06-01 14:10:43 +08:00
q.yao
b32fc41bed
[Refactor][API2.0] Api refactor2.0 ( #529 )
...
* [refactor][API2.0] Add onnx export and jit trace (#419 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add kwargs
* remove comment
* better pipeline manager
* remove print
* [Refactor][API2.0] Api partition calibration (#433 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add partition
* move calibration
* Better create_calib_table
* better deploy
* add kwargs
* remove comment
* better pipeline manager
* rename api, remove reduant variable, and misc
* [Refactor][API2.0] Api ncnn openvino (#435 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add ncnn api
* finish ncnn api
* add openvino support
* add kwargs
* remove comment
* better pipeline manager
* merge fix
* merge util and onnx2ncnn
* fix docstring
* [Refactor][API2.0] API for TensorRT (#519 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add partition
* move calibration
* Better create_calib_table
* better deploy
* add kwargs
* remove comment
* Add tensorrt API
* better pipeline manager
* add tensorrt new api
* remove print
* rename api, remove reduant variable, and misc
* add docstring
* [Refactor][API2.0] Api ppl other (#528 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add kwargs
* Add new APIS for pplnn sdk and misc
* remove comment
* better pipeline manager
* merge fix
* update tools/onnx2pplnn.py
* rename function
2022-05-31 09:18:18 +08:00
HinGwenWoong
aabab46d8a
[Enhancement] Improve MMDeploy Regression test ( #425 )
...
* make regression test as a module under project
* using `--codebase` instead of `--deploy-yml`
* Improve doc for `--codebase`
* Add shorter arg `-p` for `--performance`
* make `checopoint-dir` as an arg for the script
* Gen error log when convert fail.
* Improve res code for testing
* Doc add sdk test detail
* Doc add env setup
* Fix lint
* Fix doc lint
* Improve model path in report
* Improve report title
* Improve report checkpoint path
* Fix lint
* move test yaml under `tests/regression`
* Improve the test yaml path
* Fix lint
* Improve doc
* make func `update_report` code better
* move doc to new location
* Fix arg
* Update arg details
* Use cpu when openvino and onnxruntime cpu package
* Fix word
* Fix func of openpyxl 3.0.9
* Add some info
* Fix lint
* Fix filename
* Fix doc link
* Fix dir name with space when is not sdk
* Add args `--models` for test specific model(s)
* not saving report when no model in codebase when using `--models`
* Fix doc
* Fix lint
* Add table for metric in doc
* Improve table for doc
* Using `None` install of `['all']`
* Improce doc
* set device type properly
* Increate popen bufsize
* Add `precision_type` in `work-dir`
* Fix popen stuck
* Fix lint
* Fix lint
* Fix popen stuck by using file handler
* Make metric dataset as a list
* Update mmseg.yml
* Remove 'FPS' in the report
* Update do_regression_test.md
* Improve log
* Fix codespell
* Fix doc
* ncnn only save `xxx.param` as checkpoint name in the report
Co-authored-by: maningsheng <mnsheng@yeah.net>
2022-05-27 17:08:32 +08:00
RunningLeon
32482e76fd
compat mmpose v0.26 ( #518 )
2022-05-27 10:58:26 +08:00
tpoisonooo
ea0a9e5d43
feat(tools/deploy.py): support ncnn quantization ( #476 )
...
* feat(tools): add onnx2ncnn_quant_table
* feat(tools): add quantization image dataset
* feat(tools): add image dataset
* feat(tools/deploy.py): support quant
* fix(CI): lint
* fix(CI): format
* docs(zh_cn): add quantization usage
* docs(zh_cn): add benchmark
* feat(tools): add onnx2ncnn_quant_table
* docs(zh_cn): add more test result
* CI(github): add quant script
* CI(.github/scripts): add test quant
* fix(CI): remove pushd and popd
* feat(CI): debug
* fix(CI): path error
* fix(CI): fix path
* fix(CI): install wget
* fix(CI): review advices
* improvement(mmdeploy): review advice
* fix(tools): rename to onnx2ncnn_quant_table.py
* improvement(tools): rename file
* improvement(test): remove useless
* fix(tools/quant_image_dataset): remove loadFile in test.pipeline
* docs(quantization.md): update description
* fix(CI): protobuf version
* fix(CI): pip install
* docs(quantization): review advice
* fix(CI): revert mmcv version
* fix(CI): udpate pb version
* fix(CI): update
2022-05-26 19:53:56 +08:00
Yue Zhou
89204d16ce
[Feature] Support two stage rotated detector in MMRotate ( #473 )
...
* upload
* add fake_multiclass_nms_rotated
* delete unused code
* align with pytorch
* Update delta_midpointoffset_rbbox_coder.py
* rewrite atan2
* Update bbox_nms.py
2022-05-25 13:41:16 +08:00
RunningLeon
4f49763c28
fix mmseg twice resize ( #480 )
...
* fix mmseg twich resize
* remove comment
2022-05-25 09:52:42 +08:00
q.yao
e3a8baac4c
[Enhancement] Support RotatedRetinaNet TensorRT ( #422 )
...
* add rotated nms trt plugin
* fix ops output shape
* rebase
* fix lint
* add fp16, benchmark
* format docs
* remove unused definition, add ut
* add docs
* update docs
* add doc
2022-05-24 10:34:22 +08:00
q.yao
4710ab910d
[Refactor][tools] Add prebuild tools. ( #347 )
...
* move to lib
* optional import pytorch rewriter
* reduce torch dependancy of tensorrt export
* remove more mmcv support
* fix pytest
* remove mmcv logge
* Add `mmdeploy.utils.logging`
* Improve the common of the `get_logger`
* Fix lint
* onnxruntim add try catch to import wrapper if pytorch is available
* Using `mmcv.utils.logging` in all files under `mmdeploy/codebase`
* add __init__
* add prebuild tools
* support windows
* for comment
* exit if failed
* add exist
* decouple
* add tags
* remove .mmdeploy_python
* read python version from system
* update windows config
* update linux config
* remote many
* better build name
* rename python tag
* fix pyhon-tag
* update window config
* add env search
* update tag
* fix build without CUDA_TOOLKIT_ROOT_DIR
Co-authored-by: HinGwenWoong <peterhuang0323@outlook.com>
2022-05-23 17:15:09 +08:00
AllentDan
9eee75b98f
More ocr models ( #446 )
...
* rewrite sync batchnorm
* export panet and psenet
* resolution
* align fp16 for panet
* refine codes
* enable satrn for trt
* refine docs
* docstring
* docstring
* add ut and refine codes
* fix ut
* resolve comments and move panet-fp16 to doc
* remove ut
* refine ut
* resolve comments
* use size instead of img_scale
* use size of MultiScaleAug
Co-authored-by: dongchunyu.vendor <dongchunyu@pjlab.org.cn>
2022-05-20 21:52:36 +08:00
q.yao
a4b7bced55
[Feature] Support reppoints TensorRT ( #457 )
...
* Support reppoints tensorrt
* add ut and docs
* update zh_cn documents
* update document
2022-05-18 11:54:45 +08:00
RunningLeon
843d3c9519
Rewrite mmcls gap ( #478 )
...
* rewrite mmcls gap to replace view with flatten
* add test
* update
2022-05-17 19:57:38 +08:00
RunningLeon
6eb83a9daa
fix import mmrazor error in mmcls test ( #467 )
2022-05-13 14:15:39 +08:00