Giter Site home page Giter Site logo

mmdetection-test's Introduction

MMDetection-NPU测试用例

说明:

本仓库主要包含MMDetection相关的测试用例,用于验证MMDetection中的模型在Ascend NPU上的精度和性能。

使用方式:

  • 准备环境:

    在执行测试用例前,请按照以下指导文档正确安装mmcv和mmdetection:

    mmcv: https://mmcv.readthedocs.io/zh_CN/latest/get_started/build.html#id6

    mmdetection: https://mmdetection.readthedocs.io/zh_CN/latest/get_started.html#mmdetection

    请均按照文档命令使用源码安装。

    注意:安装mmcv框架时,如果遇到拉取代码的git链接无法下载时,可使用以下指令下载:

    git pull https://github.com/open-mmlab/mmcv.git
    
  • 准备数据:

    需要准备的数据主要为mmdetection使用真实用例执行时的dump数据,当前下载目录:

    90.90.192.17: /home/syl/code/mmdet-test/data
    

    将下载的data文件夹替换mmdetection-test/data 目录:

    文件结构如下所示:

    mmdetection-test
        ├── data
        ├── LICENSE
        ├── main.py
        ├── pytest.ini
        ├── README.md
        ├── requirements.txt
        ├── testcase
        └── utils
    
  • 测试用例执行方式:

    python3 main.py --device=910B  --scope=acc                   # 测试所有用例精度.
    
  • 参数:

    名称 可选项 默认值 作用
    scope ["acc", "prof", "all", "single"] all 指定测试精度/性能/单个用例.
    device ["UNKNOWN", "910B", "910ProB", "910A"] UNKNOWN 指定当前NPU设备的名称,用于和相同设备对比性能.
    case_id int 0 指定测试用例的ID值, 只在scope为single时生效.

已测试module :

backbones:

case_id module名称 性能 精度
0 ResNet

necks:

case_id module名称 性能 精度
1 FPN

heads:

case_id module名称 性能 精度 说明
2 Shared2FCBBoxHead
3 SingleRoIExtractor 当前版本不支持,计划330支持
4 SSDHead
5 RetinaHead
6 YOLOV3Head
7 YOLOXHead
8 CenterNetHead
9 FCOSHead
10 SOLOV2Head

others:

case_id module名称 性能 精度
11 CrossEntropyLoss
12 MaxIoUAssigner

mmdetection-test's People

Contributors

ginray avatar

Watchers

 avatar  avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.