Commit Graph

642 Commits (441d0e27037f76dc4686783ae0806814aa664889)
 

Author SHA1 Message Date
AllentDan 441d0e2703
fix two stage batch dynamic (#1046) 2022-09-14 17:48:34 +08:00
AllentDan 615668ec63
update api in doc (#1021) 2022-09-14 17:47:26 +08:00
q.yao e912e86d24
BUG P0 (#1044) 2022-09-14 17:20:11 +08:00
Chen Xin b87afb9ebb
fix mmdeploy builder on windows (#1018)
* fix mmdeploy builder on windows

* add pyyaml

* fix lint
2022-09-09 15:03:55 +08:00
q.yao f2bc808fe3
New issue template (#1007)
* update bug report

* update issue template

* update bug-report
2022-09-07 16:00:37 +08:00
tpoisonooo 8e9578c04d
fix(CI): update link checker (#1008) 2022-09-07 15:59:48 +08:00
lvhan028 b310ba8d3e
bump version to v0.8.0 (#1009) 2022-09-07 10:53:50 +08:00
Li Zhang b602356181
[Fix] Add isolated option for TorchScript SDK backend (#1002)
* add option for TorchScript SDK backend

* add doc

* format
2022-09-06 12:24:07 +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
Chen Xin a0fb3be0df
support coreml (#760)
* sdk inference

* fix typo

* fix typo

* add convert things

* fix missling name

* add cls support

* add more pytorch rewriter

* add det support

* support det wip

* make Model export model_path

* fix nms

* add output back

* add docstring

* fix lint

* add coreml build action

* add zh docs

* add coreml backend check

* update ci

* update

* update

* update

* update

* update

* fix lint

* update configs

* add return value when error occured

* update docs

* update docs

* update docs

* fix lint

* udpate docs

* udpate docs

* update

Co-authored-by: grimoire <streetyao@live.com>
2022-09-05 19:55:47 +08:00
munhou 3d092bfb39
fix mmdet showresult (#999)
* fix mmdet showresult

* Consider compatibility

* mmdet showresult add *args

* Revert "mmdet showresult add *args"

This reverts commit 82265a31cf.

Co-authored-by: whhuang <whhuang@hitotek.com>
2022-09-05 18:27:02 +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
tpoisonooo e21cad84e0
fix(backend): disable cublaslt for cu102 (#947)
* fix(backend): disable cublaslt for cu102

* fix

* fix(backend): update

* fix(tensorrt/util.py): add find cuda version

* fix

* fix(CI): first use cmd to get cuda version

* docs(tensorrt/utils.py): update docstring
2022-09-05 14:52:43 +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
Hongyi Xiang 9918d29212
add missing sqrt for PAAHead's score calculation (#984)
Co-authored-by: xianghongyi1 <xianghongyi1@sensetime.com>
2022-09-02 18:25:50 +08:00
tpoisonooo 9b3db48de9
fix(CI): ort test all pipeline (#985) 2022-09-02 17:34:22 +08:00
tpoisonooo 306616fc34
fix(CI): typo (#983) 2022-09-02 16:06:53 +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 5874f1039b
Add CI to test full pipeline (#966)
* add mmcls full pipeline test ci

* update

* update

* add mmcv

* install torch

* install mmdeploy

* change clone with https

* install mmcls

* update

* change mmcls version

* add mmcv version

* update  mmcls version

* test sdk

* tast with imagnet

* sed pipeline

* print env

* update

* move to backend-ort ci

* install mim
2022-09-02 10:55:53 +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
RunningLeon a6ed7eddc5
fix mmroate (#976) 2022-09-02 10:45:49 +08:00
tpoisonooo 38bcc43256
CI(tools/scripts): add submodule init and update (#977) 2022-09-02 10:28:11 +08:00
Chen Xin 8a40f195bc
remove Release dir in mmdeploy package (#960) 2022-08-31 15:48:42 +08:00
hanrui1sensetime ac9db5b4e7
[Fix] Close onnx optimizer for ncnn (#961)
* close onnx optimizer for ncnn

* fix docformatter

* fix lint
2022-08-31 15:48:08 +08:00
q.yao eed69b14e2
fix ci (#964) 2022-08-31 14:36:15 +08:00
gy77 638e33a964
add_multi_label_postprocess (#950)
* 'add_multi_label_postprocess'

* fix pre-commit

* delete partial_sort

* delete idx

* delete num_classes and num_classes_

* Fix right brackets and spelling errors in lines 19 and 20

Co-authored-by: gaoying <gaoying@xiaobaishiji.com>
2022-08-30 19:02:42 +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
tpoisonooo f7e0905e95
docs(convert_model): update description (#956) 2022-08-30 10:37:07 +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
Li Zhang 6c2dde1cf8
[Feature] TorchScript SDK backend (#890)
* WIP SDK torchscript support

* support detection task

* make torchvision optional

* force link torchvision if enabled

* support torch-1.12

* fix export & sync cuda stream

* hide internal classes

* handle error

* set `MMDEPLOY_USE_CUDA` when CUDA is enabled
2022-08-29 18:01:18 +08:00
Chen Xin c04dcd2c62
Support risc-v platform (#910)
* add ppl.nn riscv engine

* update ppl.nn riscv engine

* udpate riscv service (ncnn backend)

* update _build_wrapper for ncnn

* fix build

* fix lint

* update default uri

* update file structure & add cn doc

* remove copy input data

* update docs

* remove ncnn server

* fix docs

* update zh doc

* update toolchain

* remove unused

* update doc

* update doc

* update doc

* rename cross build dirname

* add riscv.md to build_from_source.md

* update cls model

* test ci

* test ci

* test ci

* test ci

* test ci

* update ci

* update ci
2022-08-26 13:54:40 +08:00
tpoisonooo 4534598056
improvement(installation): add script install mmdeploy (#919)
* feat(tools): add build ubuntu x64 ncnn

* ci(tools): add ncnn auto install

* fix(ci): auto install ncnn

* fix(tools): no interactive

* docs(build): add script build

* CI(ncnn): script install ncnn

* docs(zh_cn): fix error os

* fix

* CI(tools/script): test ort install passed

* update

* CI(tools): support pplnn

* CI(build): add pplnn

* docs(tools): update

* fix

* CI(tools): script install torchscript

* docs(build): add torchscript

* fix(tools): clean code and doc

* update

* fix(CI): requirements install failed

* debug CI

* update

* update

* update

* feat(tools/script): support user specify make jobs

* fix(tools/script): fix build pplnn with cuda

* fix(tools/script): torchscript add tips and simplify install mmcv

* fix(tools/script): check nvcc version first

* fix(tools/scripts): pplnn checkout

* fix(CI): add simple check install succcess

* fix

* debug CI

* fix

* fix(CI): pplnn install mis wheel

* fix(CI): build error

* fix(CI): remove misleading message
2022-08-26 13:53:53 +08:00
RunningLeon b1e7579bdd
add cflags explicitly in ci (#945) 2022-08-25 13:54:42 +08:00
tpoisonooo 21775ce584
docs(mmdet3d): give detail model path (#940) 2022-08-24 15:11:58 +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
AllentDan e1aabf5837
[Doc] How to write a customized TensorRT plugin (#290)
* first edition

* fix lint

* add 06, 07

* resolve comments

* update index.rst

* update title

* update img
2022-08-22 14:00:28 +08:00
RunningLeon 1f8d889b36
set test_mode for mmdet (#920)
* fix

* update
2022-08-19 10:55:41 +08:00
tpoisonooo a6e07dac2f
improvement(cmake): simplify build option and doc (#832)
* improvement(cmake): simplify build option

improvement(cmake): convert target_backends with directory

* fix(dockerfile): build error

* fix(CI): circle CI

* fix(docs): snpe and cmake option

* fix(docs): revert update cmake

* fix(docs): revert

* update(docs): remove useless
2022-08-19 10:54:09 +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 3e7e80bcbc
support multi-batch test in profile tool (#868)
* test batch profile with resnet pspnet yolov3 srcnn

* update doc

* update docs

* fix ut

* fix mmdet

* support batch mmorc and mmrotate

* fix mmcls export to sdk

* resolve comments

* rename to fix #819

* fix conflicts with master
2022-08-17 15:01:06 +08:00
AllentDan 9fbfdd2178
add device backend check (#886)
* add device backend check

* safe check

* only activated for tensorrt and openvino

* resolve comments
2022-08-16 17:20:29 +08:00
tpoisonooo 3fa15822b1
docs(zh_cn): add architect (#882)
* docs(zh_cn): add architect

docs(en): add architect

fix(docs): readthedocs index

* docs(en): update architect.md

* docs(README.md): update

* docs(architecture): fix review advices
2022-08-15 15:16:20 +08:00
lvhan028 91a060fdbb
fix mmdeploy_pplnn_net build error when target device is cpu (#896) 2022-08-15 14:51:28 +08:00
tpoisonooo 127125f641
docs(project): sync en and zh docs (#842)
* docs(en): update file structure

* docs(zh_cn): update

* docs(structure): update

* docs(snpe): update

* docs(README): update

* fix(CI): update

* fix(CI): index.rst error

* fix(docs): update

* fix(docs): remove mermaid

* fix(docs): remove useless

* fix(docs): update link

* docs(en): update

* docs(en): update

* docs(zh_cn): remove \[

* docs(zh_cn): format

* docs(en): remove blank

* fix(CI): doc link error

* docs(project): remove "./" prefix

* docs(zh_cn): fix mdformat

* docs(en): update title

* fix(CI): update docs
2022-08-15 10:18:17 +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