mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
* 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 043490075e6dbe4a8fb98e94b2b583b91fc5038d. * [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>
371 lines
12 KiB
YAML
371 lines
12 KiB
YAML
globals:
|
|
codebase_dir: ../mmsegmentation
|
|
checkpoint_force_download: False
|
|
images:
|
|
img_leftImg8bit: &img_leftImg8bit ../mmsegmentation/tests/data/pseudo_cityscapes_dataset/leftImg8bit/frankfurt_000000_000294_leftImg8bit.png
|
|
img_loveda_0: &img_loveda_0 ../mmsegmentation/tests/data/pseudo_loveda_dataset/img_dir/0.png
|
|
img_loveda_1: &img_loveda_1 ../mmsegmentation/tests/data/pseudo_loveda_dataset/img_dir/1.png
|
|
img_loveda_2: &img_loveda_2 ../mmsegmentation/tests/data/pseudo_loveda_dataset/img_dir/2.png
|
|
img_potsdam: &img_potsdam ../mmsegmentation/tests/data/pseudo_potsdam_dataset/img_dir/2_10_0_0_512_512.png
|
|
img_vaihingen: &img_vaihingen ../mmsegmentation/tests/data/pseudo_vaihingen_dataset/img_dir/area1_0_0_512_512.png
|
|
metric_info: &metric_info
|
|
mIoU: # named after metafile.Results.Metrics
|
|
eval_name: mIoU # test.py --metrics args
|
|
metric_key: mIoU # eval OrderedDict key name
|
|
tolerance: 5 # metric ±n%
|
|
task_name: Semantic Segmentation # metafile.Results.Task
|
|
dataset: [Cityscapes, ADE20K] # metafile.Results.Dataset
|
|
convert_image: &convert_image
|
|
input_img: *img_leftImg8bit
|
|
test_img: *img_loveda_0
|
|
backend_test: &default_backend_test True
|
|
sdk:
|
|
sdk_dynamic: &sdk_dynamic configs/mmseg/segmentation_sdk_dynamic.py
|
|
|
|
onnxruntime:
|
|
pipeline_ort_static_fp32: &pipeline_ort_static_fp32
|
|
convert_image: *convert_image
|
|
backend_test: *default_backend_test
|
|
sdk_config: *sdk_dynamic
|
|
deploy_config: configs/mmseg/segmentation_onnxruntime_static-1024x2048.py
|
|
|
|
pipeline_ort_static_fp32_512x512: &pipeline_ort_static_fp32_512x512
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_onnxruntime_static-512x512.py
|
|
|
|
pipeline_ort_dynamic_fp32: &pipeline_ort_dynamic_fp32
|
|
convert_image: *convert_image
|
|
deploy_config: configs/mmseg/segmentation_onnxruntime_dynamic.py
|
|
|
|
tensorrt:
|
|
pipeline_trt_static_fp32: &pipeline_trt_static_fp32
|
|
convert_image: *convert_image
|
|
backend_test: *default_backend_test
|
|
sdk_config: *sdk_dynamic
|
|
deploy_config: configs/mmseg/segmentation_tensorrt_static-1024x2048.py
|
|
|
|
pipeline_trt_static_fp32_512x512: &pipeline_trt_static_fp32_512x512
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_tensorrt_static-512x512.py
|
|
|
|
pipeline_trt_static_fp16: &pipeline_trt_static_fp16
|
|
convert_image: *convert_image
|
|
backend_test: *default_backend_test
|
|
sdk_config: *sdk_dynamic
|
|
deploy_config: configs/mmseg/segmentation_tensorrt-fp16_static-1024x2048.py
|
|
|
|
pipeline_trt_static_int8: &pipeline_trt_static_int8
|
|
convert_image: *convert_image
|
|
backend_test: *default_backend_test
|
|
sdk_config: *sdk_dynamic
|
|
deploy_config: configs/mmseg/segmentation_tensorrt-int8_static-1024x2048.py
|
|
|
|
pipeline_trt_static_fp16_512x512: &pipeline_trt_static_fp16_512x512
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_tensorrt-fp16_static-512x512.py
|
|
|
|
pipeline_trt_dynamic_fp32: &pipeline_trt_dynamic_fp32
|
|
convert_image: *convert_image
|
|
backend_test: *default_backend_test
|
|
sdk_config: *sdk_dynamic
|
|
deploy_config: configs/mmseg/segmentation_tensorrt_dynamic-512x1024-2048x2048.py
|
|
|
|
pipeline_trt_dynamic_fp16: &pipeline_trt_dynamic_fp16
|
|
convert_image: *convert_image
|
|
backend_test: *default_backend_test
|
|
sdk_config: *sdk_dynamic
|
|
deploy_config: configs/mmseg/segmentation_tensorrt-fp16_dynamic-512x1024-2048x2048.py
|
|
|
|
pipeline_trt_dynamic_int8: &pipeline_trt_dynamic_int8
|
|
convert_image: *convert_image
|
|
backend_test: *default_backend_test
|
|
sdk_config: *sdk_dynamic
|
|
deploy_config: configs/mmseg/segmentation_tensorrt-int8_dynamic-512x1024-2048x2048.py
|
|
|
|
openvino:
|
|
pipeline_openvino_dynamic_fp32: &pipeline_openvino_dynamic_fp32
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_openvino_dynamic-1024x2048.py
|
|
|
|
pipeline_openvino_static_fp32: &pipeline_openvino_static_fp32
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_openvino_static-1024x2048.py
|
|
|
|
pipeline_openvino_static_fp32_512x512: &pipeline_openvino_static_fp32_512x512
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_openvino_static-512x512.py
|
|
|
|
ncnn:
|
|
pipeline_ncnn_static_fp32: &pipeline_ncnn_static_fp32
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_ncnn_static-512x512.py
|
|
|
|
pplnn:
|
|
pipeline_pplnn_dynamic_fp32: &pipeline_pplnn_dynamic_fp32
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_pplnn_dynamic-1024x2048.py
|
|
|
|
pipeline_pplnn_static_fp32: &pipeline_pplnn_static_fp32
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_pplnn_static-1024x2048.py
|
|
|
|
torchscript:
|
|
pipeline_ts_fp32: &pipeline_ts_fp32
|
|
convert_image: *convert_image
|
|
backend_test: False
|
|
deploy_config: configs/mmseg/segmentation_torchscript.py
|
|
|
|
|
|
models:
|
|
- name: FCN
|
|
metafile: configs/fcn/fcn.yml
|
|
model_configs:
|
|
- configs/fcn/fcn_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ts_fp32
|
|
- *pipeline_ort_dynamic_fp32
|
|
# - *pipeline_trt_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
# - *pipeline_trt_dynamic_int8
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_pplnn_dynamic_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: PSPNet
|
|
metafile: configs/pspnet/pspnet.yml
|
|
model_configs:
|
|
- configs/pspnet/pspnet_r50-d8_512x1024_80k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ts_fp32
|
|
- *pipeline_ort_static_fp32
|
|
- *pipeline_trt_static_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_pplnn_static_fp32
|
|
- *pipeline_openvino_static_fp32
|
|
|
|
- name: deeplabv3
|
|
metafile: configs/deeplabv3/deeplabv3.yml
|
|
model_configs:
|
|
- configs/deeplabv3/deeplabv3_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ts_fp32
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_pplnn_dynamic_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: deeplabv3+
|
|
metafile: configs/deeplabv3plus/deeplabv3plus.yml
|
|
model_configs:
|
|
- configs/deeplabv3plus/deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ts_fp32
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_pplnn_dynamic_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: Fast-SCNN
|
|
metafile: configs/fastscnn/fastscnn.yml
|
|
model_configs:
|
|
- configs/fastscnn/fast_scnn_lr0.12_8x4_160k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ts_fp32
|
|
- *pipeline_ort_static_fp32
|
|
- *pipeline_trt_static_fp16
|
|
- *pipeline_pplnn_static_fp32
|
|
- *pipeline_openvino_static_fp32
|
|
|
|
- name: UNet
|
|
metafile: configs/unet/unet.yml
|
|
model_configs:
|
|
- configs/unet/fcn_unet_s5-d16_4x4_512x1024_160k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ts_fp32
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_pplnn_dynamic_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: ANN
|
|
metafile: configs/ann/ann.yml
|
|
model_configs:
|
|
- configs/ann/ann_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_static_fp32
|
|
- *pipeline_trt_static_fp16
|
|
|
|
- name: APCNet
|
|
metafile: configs/apcnet/apcnet.yml
|
|
model_configs:
|
|
- configs/apcnet/apcnet_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
|
|
- name: BiSeNetV1
|
|
metafile: configs/bisenetv1/bisenetv1.yml
|
|
model_configs:
|
|
- configs/bisenetv1/bisenetv1_r18-d32_4x4_1024x1024_160k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: BiSeNetV2
|
|
metafile: configs/bisenetv2/bisenetv2.yml
|
|
model_configs:
|
|
- configs/bisenetv2/bisenetv2_fcn_4x4_1024x1024_160k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: CGNet
|
|
metafile: configs/cgnet/cgnet.yml
|
|
model_configs:
|
|
- configs/cgnet/cgnet_512x1024_60k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: EMANet
|
|
metafile: configs/emanet/emanet.yml
|
|
model_configs:
|
|
- configs/emanet/emanet_r50-d8_512x1024_80k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: EncNet
|
|
metafile: configs/encnet/encnet.yml
|
|
model_configs:
|
|
- configs/encnet/encnet_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: ERFNet
|
|
metafile: configs/erfnet/erfnet.yml
|
|
model_configs:
|
|
- configs/erfnet/erfnet_fcn_4x4_512x1024_160k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: FastFCN
|
|
metafile: configs/fastfcn/fastfcn.yml
|
|
model_configs:
|
|
- configs/fastfcn/fastfcn_r50-d32_jpu_aspp_512x1024_80k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: GCNet
|
|
metafile: configs/gcnet/gcnet.yml
|
|
model_configs:
|
|
- configs/gcnet/gcnet_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
|
|
- name: ICNet
|
|
metafile: configs/icnet/icnet.yml
|
|
model_configs:
|
|
- configs/icnet/icnet_r18-d8_832x832_80k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_static_fp32
|
|
- *pipeline_trt_static_fp16
|
|
- *pipeline_openvino_static_fp32
|
|
|
|
- name: ISANet
|
|
metafile: configs/isanet/isanet.yml
|
|
model_configs:
|
|
- configs/isanet/isanet_r50-d8_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: OCRNet
|
|
metafile: configs/ocrnet/ocrnet.yml
|
|
model_configs:
|
|
- configs/ocrnet/ocrnet_hr18s_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: PointRend
|
|
metafile: configs/point_rend/point_rend.yml
|
|
model_configs:
|
|
- configs/point_rend/pointrend_r50_512x1024_80k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: Semantic FPN
|
|
metafile: configs/sem_fpn/sem_fpn.yml
|
|
model_configs:
|
|
- configs/sem_fpn/fpn_r50_512x1024_80k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: STDC
|
|
metafile: configs/stdc/stdc.yml
|
|
model_configs:
|
|
- configs/stdc/stdc1_in1k-pre_512x1024_80k_cityscapes.py
|
|
- configs/stdc/stdc2_in1k-pre_512x1024_80k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_dynamic_fp32
|
|
- *pipeline_trt_dynamic_fp16
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_openvino_dynamic_fp32
|
|
|
|
- name: UPerNet
|
|
metafile: configs/upernet/upernet.yml
|
|
model_configs:
|
|
- configs/upernet/upernet_r50_512x1024_40k_cityscapes.py
|
|
pipelines:
|
|
- *pipeline_ort_static_fp32
|
|
- *pipeline_trt_static_fp16
|
|
- name: Segmenter
|
|
metafile: configs/segmenter/segmenter.yml
|
|
model_configs:
|
|
- configs/segmenter/segmenter_vit-s_mask_8x1_512x512_160k_ade20k.py
|
|
- configs/segmenter/segmenter_vit-s_linear_8x1_512x512_160k_ade20k.py
|
|
pipelines:
|
|
- *pipeline_ort_static_fp32_512x512
|
|
- *pipeline_trt_static_fp32_512x512
|
|
- *pipeline_openvino_static_fp32_512x512
|
|
- *pipeline_ncnn_static_fp32
|
|
- *pipeline_ts_fp32
|