Commit Graph

645 Commits (a1a19f0fc1a8e902a2f770cc22a35035817bc043)
 

Author SHA1 Message Date
tpoisonooo 41ddcafbb4
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
2022-07-08 09:21:32 +08:00
RunningLeon 22733c2838
fix passing wrong backend type (#719) 2022-07-07 22:34:34 +08:00
RunningLeon 585e34b229
fix wrong exit code in pipeline_manager (#715)
* fix exit

* change to general exit errorcode=1
2022-07-07 15:00:55 +08:00
hanrui1sensetime d57cf85e1d
[Fix] Fix android build (#698)
* fix android build

* fix cmake

* fix url link
2022-07-07 14:29:45 +08:00
RunningLeon 69c38b9f74
fix passing img_metas for mmdet3d (#707) 2022-07-05 17:50:09 +08:00
RunningLeon 029ed69db0
fix passing img_metas to pytorch2onnx for mmedit (#700) 2022-07-05 11:48:33 +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
RunningLeon 5195ff9388
bump version to 0.6.0 (#680)
* bump vertion to 0.6.0

* update version
2022-06-30 16:43:44 +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
zambranohally e49e8a10ed
Update get_started.md (#676)
Fix backend model assignment
2022-06-29 19:04:58 +08:00
zambranohally 3bc766e6f8
Update get_started.md (#675)
Fix backend model assignment
2022-06-29 19:02:20 +08:00
Chen Xin c792d06706
fix export sdk info of input shape (#667) 2022-06-29 15:19:39 +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
lvhan028 b1f156ad0a
update introduction.png (#674) 2022-06-29 14:18:36 +08:00
Bryan Glen Suello 0f0f676d1f
Fixed tensorrt plugin not found in Windows (#672) 2022-06-29 13:06:14 +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
hanrui1sensetime 051e1577a3
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
2022-06-29 11:02:08 +08:00
RunningLeon 5fce1e8f8d
update docs (#624) 2022-06-29 10:15:48 +08:00
RunningLeon f568fe7da5
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
2022-06-28 19:46:58 +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
tripleMu 05cafab723
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>
2022-06-27 19:43:09 +08:00
Li Zhang 82a74d8717
simplify `Expand` (#617) 2022-06-27 18:42:57 +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
Ryan_Huang 7e8d6eb50f
' -> " (#655)
same with https://github.com/open-mmlab/mmdeploy/pull/654
2022-06-27 17:32:39 +08:00
Ryan_Huang 49608b4289
' -> " (#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
2022-06-27 17:32:14 +08:00
RunningLeon fa034e042b
fix shufflenetv2 with trt (#645)
* fix shufflenetv2 and pspnet

* fix ci

* remove print
2022-06-27 10:48:48 +08:00
q.yao ae47e9d188
Reduce mmcls version dependency (#635) 2022-06-27 09:41:36 +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
Chen Xin f30eacff17
[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
2022-06-21 19:19:44 +08:00
Ryan_Huang 9c8168f39a
make -install -> make install (#621)
change `make -install` to `make install`

https://github.com/open-mmlab/mmdeploy/issues/618
2022-06-20 14:46:56 +08:00
Li Zhang ac0b52f12a
add dim param for `Tensor::Squeeze` (#603) 2022-06-17 14:06:35 +08:00
Li Zhang a822ba7330
add `MMDEPLOY_TASKS` & simplify demo cmake (#606)
* add `MMDEPLOY_TASKS` & simply cmake

* remove debug log
2022-06-17 13:56:45 +08:00
RunningLeon 296412fd08
[Docs] Replace markdownlint with mdformat and configure myst-parser (#610)
* replace mdlint with mdformat

* set myst_heading_anchor for doc

* update precommit yml

* format md files
2022-06-17 09:19:10 +08:00
RunningLeon 3e594b28f6
Fix index.rst and add check ci (#607)
* move doc link check to lint ci and add index.rst file check

* fix filename
2022-06-16 18:45:04 +08:00
hanrui1sensetime ef388032d1
[Fix] adapt mmocr different versions (#604)
* adapt mmocr different versions

* fix lint

* fix runningleon comment
2022-06-16 17:44:16 +08:00
hanrui1sensetime 3f131138d8
fix ocr textdet (#605) 2022-06-16 11:11:18 +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
Yosshi999 6c7859510d
Support compute capability 87 archs such as Orin (#601) 2022-06-15 19:01:36 +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
Li Zhang 86a50e343a
[Fix] Fix demo link (#588)
* fix demo link

* update urls
2022-06-13 11:07:34 +08:00
Liu Qingchen e84bc30f4a
fix a typo in Build for Jetson (#586)
fix a typo described in issue #585
2022-06-10 13:37:15 +08:00
lvhan028 4335313fb7
bump version to v0.5.0
bump version to v0.5.0
2022-06-09 13:22:25 +08:00
Chen Xin cb09d2238b
fix mmcls-c-demo (#580)
Co-authored-by: irexyc <chenxin2@sensetime.com>
2022-06-09 11:45:57 +08:00
RunningLeon b29a8a020a
fix build ci not triggerred (#578)
* fix ci

* ignore scripts

* remove pycuda from requirements

* fix trt apis
2022-06-09 11:19:05 +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
lvhan028 e69b7a5838
[docs] fix readthedocs (#577)
* Fix readthedocs issues

* fix opencv-python to 4.5.4.60

* update according to reviewer comments
2022-06-08 20:39:34 +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