Commit Graph

253 Commits (9c6b4fedc8adf543708b42278610315f8fb43971)
 

Author SHA1 Message Date
tripleMu 6fd50af6ab
Export onnx for model only (#361)
* Support export onnx for model only

* Fix

* Fix
2022-12-13 15:06:59 +08:00
tianlei e7ff6fcbf0
Add torch1.13 checking in CI (#363) 2022-12-12 15:49:54 +08:00
RunningLeon dcf73091e5
add mmyolo regression test yml (#359) 2022-12-12 10:27:21 +08:00
Range King 21be0c4c68
Add some docstring (#360) 2022-12-12 10:23:58 +08:00
HinGwenWoong 2bf502b13c
[Improvement] Improve doc `custom_dataset` (#336)
* Improve doc

* Improve doc

* Add config

* Add config

* Add config

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc index

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Add deploy demo

* Improve doc

* Fix lint

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve code

* Improve code

* Improve code

* Improve code

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc
2022-12-12 10:16:03 +08:00
hanrui1sensetime a20d3c14fc
fix mmyolo unittest scope bug (#351) 2022-12-08 11:45:54 +08:00
Haian Huang(深度眸) e57edb65c6
fix pad_param error (#354) 2022-12-08 11:45:22 +08:00
Yue WANG 王跃 0e28babbee
[Docs] Update video tutorials in article.md (#350)
* Update article.md

* Update article.md
2022-12-07 18:52:09 +08:00
HinGwenWoong 91489bf065
[Feature] Add deploy demo (#343)
* [Feature] Add deploy demo

* [Feature] Add deploy demo

* [Feature] Add deploy demo

* [Feature] Add deploy demo
2022-12-06 18:53:26 +08:00
Haian Huang(深度眸) 92fd724e87
[Enhancement] Optimize the vis results of large images in `debug` mode (#346)
* Optimize the vis results of large images in debug mode

* to test

* to test

* remove

* update

* update
2022-12-06 18:47:43 +08:00
HinGwenWoong 78a23ca8b9
[Fix] Fix `dataset_analysis` ProgressBar incorrect when `RepeatDataset` (#338)
* Fix ProgressBar incorrect when RepeatDataset

* Fix lint

* Improve doc

* Improve doc

* Improve import

* Fix hard code

* Fix hard code
2022-12-06 10:21:41 +08:00
HinGwenWoong 46b2494453
[Improvement] Improve args for `browse_dataset` and supprot `RepeatDataset` (#340)
* Improve doc and Fix bug

* Fix hard code

* Fix hard code

* Update browse_dataset.py
2022-12-06 10:19:49 +08:00
Nioolek 12e8f0ea07
fix twice head inference bug (#342) 2022-12-06 10:18:58 +08:00
HinGwenWoong e3d95069ef
[Improvement] Improve args for `tools/analysis_tools/optimize_anchors.py` (#339)
* Improve args

* Improve doc

* Improve doc

* Improve doc
2022-12-04 17:18:33 +08:00
CaiLiYuan 9985e0bc08
Update yolov7_l_syncbn_fast_8x16b-300e_coco.py (#333)
change:model->bbox_head->head_module->dict(num_classes=num_classes)
2022-12-02 16:31:27 +08:00
Haian Huang(深度眸) 27487fd587
Merge pull request #331 from open-mmlab/dev
Bump version to v0.2.0
2022-12-01 11:14:19 +08:00
Haian Huang(深度眸) 4e8dac5fff
add changelog of v0.2.0 (#323)
* add changelog v0.2.0

* update

* update

* update

* update

* update
2022-12-01 10:54:44 +08:00
MingJian.L 1f06f4f594
[Improvement] `browse_dataset.py` (#304)
* Create browse_transform.py

Upgrate the brow_transform
对pipeline中的transform过程实现了可视化,以及可以将transform中key参数的变化打印出来。

* Update browse_dataset.py

* Delete browse_transform.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

修改了215行result_i = [result['dataset_sample'] for result in intermediate_imgs]通过lint

* fix some error

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2022-12-01 10:26:00 +08:00
Range King 98e6fcccc7
[Docs] Update EN `how_to.md` with the examples of mim (#330)
* Update en how_to

* fix typos in zh_cn how_to
2022-11-30 19:28:42 +08:00
Range King 7431231d87
[Docs] Add docs about how to use mim to run scripts across libraries (#321)
* Update how_to.md

* Update how_to.md

* Update useful_tools.md

* Update useful_tools.md

* Update how_to.md
2022-11-30 19:03:02 +08:00
HinGwenWoong b8afd0055d
[DOC] Add custom dataset guide (#306)
* Add doc

* Add doc

* Add doc

* Add doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve config

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc config

* Improve doc config

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc
2022-11-30 18:58:43 +08:00
Haian Huang(深度眸) e85ddeac0d
[Feature] Support YOLOv7 P6 training (#310)
* support p6 train

* fix bug

* add readme

* add link
2022-11-30 18:45:08 +08:00
hanrui1sensetime 4a8699d6fe
[Fix] Enable switch to deploy when create pytorch model in deployment. (#324)
* support switch_to_deploy when deploy

* fix docformatter

* fix README for yolox shape
2022-11-29 18:23:12 +08:00
AllentDan 68c9fd4745
[Feature] Support YOLOv5 deployment at RV1126 device (#262)
* add yolov5 head rewrite and rknn config

* type hint

* remove useless

* refactor config
2022-11-29 16:54:08 +08:00
Range King 3d04ca2f79
[Docs] Add `contributing.md` and `code_style.md` (#322)
* Add zh_cn community docs

* Update index.rst

* Create contributing.md

* Create code_style.md

* Update index.rst
2022-11-29 10:10:29 +08:00
Range King 2354302c94
[Improvement] Add docstring and update RTMDet graph (#317)
* add docstring

* Update yolov5_coco.py

* Update rtmdet_description.md

* Update batch_yolov7_assigner.py
2022-11-28 17:13:20 +08:00
Range King 55da55eb09
Update mix_img_transforms.py (#319) 2022-11-28 09:58:00 +08:00
Range King 60a515f41f
[Enhance] Add option of json output in `test.py` (#316)
* add json args

* Update how_to.md

* Update test.py

* Update how_to.md

* Update how_to.md
2022-11-26 19:59:37 +08:00
Range King 5efa56b3b5
[Docs] Update video tutorials in `article.md` (#318)
* Update article.md

* Update article.md

* Update article.md
2022-11-26 19:58:41 +08:00
AZhou 4b79e4f983
[Feature] Add condition for extract coco datasets (#286)
* add condition for extract coco datasets

* fix lint error

* guarantee images be extracted definitely has annotation info

* reset num_img default value
2022-11-26 16:19:04 +08:00
Xin Li 7440a68e51
[Docs] Deployment translation (#289)
* translate deployment docs

* translate deployment docs part 2

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/deploy/yolov5_deployment.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/deploy/basic_deployment_guide.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/deploy/basic_deployment_guide.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/deploy/basic_deployment_guide.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* debug link representation in readthedocs

* debug link representation in readthedocs

* debug link representation in readthedocs

Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2022-11-26 10:07:50 +08:00
HinGwenWoong daae790075
[Feature] Add split COCO dataset script (#311)
* Add coco_sqlit script

* rename file

* Improve code

* Improve code

* Improve coding

* Improve file structure

* ratios normalize
2022-11-26 09:27:30 +08:00
hanrui1sensetime 15224b7816
[Fix] Fix mmyolo int8 deploy config (#315)
* fix mmyolo int8 deploy config

* fix lint
2022-11-26 09:26:40 +08:00
HinGwenWoong 285c395066
[Improvement] Add output `class_with_id.txt` after labelme2coco (#313)
* Add output class_with_id.json

* Add `--class-id-json`

* Improve coding, using txt and saving txt

* Improve coding
2022-11-25 16:50:50 +08:00
HinGwenWoong e3f1cf93a6
[Improvement] Add `--class-name` for filter when save to labelme label. (#314)
* Add `--class-name` for filter when save to labelme label.

* Improve code

* Add check class

* Improve coding

* Improve coding
2022-11-25 16:48:59 +08:00
Haian Huang(深度眸) 9221499af4
[Feature] Support grad-based cam and grad-free cam (#234)
* support cam

* update

* update

* done

* fix lint

* add docstr

* add doc

* update

* update

* FIX
2022-11-25 16:45:40 +08:00
tripleMu 1045b41b68
[Deploy] MMYOLO model convert to onnx for deployment. (#279)
* Fromat code

* Support ONNXRUNTIME

* Support mmyolo model convert to onnx for deploy.

* Same as dev branch

* Support yolox focus rewrite

* Support GConv Focus

* Update mmyolo/easydeploy/backbone/focus.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Add TensorRT build/infer Wrapper

* Add image_demo for deploy model

* Fix

* Merge dev

* Remove image-demo

* Roll back to dev

* Support model switch to deploy

* Remove --deploy

* Add new deploy method

* Format code and add doc

* Move md to project

* add readme and readme_zh to easy_deploy

* Update projects/easydeploy/README_zh-CN.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update projects/easydeploy/README.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: xin-li-67 <williamlee.xin@gmail.com>
Co-authored-by: Xin Li <7219519+xin-li-67@users.noreply.github.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-11-23 10:43:41 +08:00
HinGwenWoong b1e478a8bd
[Feature] Add `labelme2coco` script (#308)
* Add label2coco script

* Add label2coco script

* Add label2coco script

* Add label2coco script

* Improve script

* Improve script
2022-11-23 10:32:52 +08:00
kitecats 748f151886
[Refactor] Refactor some of the functionality of `dataset_analysis` (#294)
* Refactor some of the functionality of data_analysis

* Remove redundant code in dataset_analysis.py

* Update tools/analysis_tools/dataset_analysis.py

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* simplified code

* add docstring and simplify code

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-11-22 11:21:35 +08:00
HinGwenWoong 47bb3ce408
[Improvement] Improve browse coco json script args (#309)
* Improve browse coco json script

* Improve doc

* Improve script

* Improve doc

* Improve doc

* Improve doc
2022-11-22 10:51:42 +08:00
HinGwenWoong 1773439b83
[Fix] Fix scale factor order of `LetterResize` and `YOLOv5KeepRatioResize` (#305)
* Fix bug

* Fix bug

* Fix lint

* Fix scale from hw -> wh

* Fix scale from hw -> wh
2022-11-21 19:44:33 +08:00
Range King 82d1680b4d
[Improvement] Add mosaic9 graph in docstring (#307)
* Add mosaic9 graph in docstring

* fix typos
2022-11-21 19:13:52 +08:00
Haian Huang(深度眸) 573ff033e4
[Feature] Support YOLOv7 P5 training (#243)
* support yolov7

* update dev

* update headmodule and convert

* update

* fix pipeline

* fix loss

* fix optimizer parameter groups

* refactor mosaic9

* refactor optim

* refactor loss

* refactor

* refactor

* refactor new

* support yolov7x

* refine

* support tiny

* refactor model

* refactor model

* support yolov7x inference

* support yolov7-tiny inference

* support yolov7-e inference

* refactor

* support yolov7-tiny train

* add docstr

* fix merge error

* fix merge error

* fix merge error

* fix lint

* fix lint

* fix lint

* fix UT

* update

* update
2022-11-21 17:30:19 +08:00
HinGwenWoong 5e0599c825
[Feature] Add flag for output labelme label file in `image_demo` (#288)
* Add flag for labelme

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Not save `imageData`

* preprocess score

* preprocess score

* Fix lint
2022-11-20 15:37:58 +08:00
LinXiaoZheng 12a1e8a984
[Docs] Fix config.md (#297)
* fix_config.md

* Delete yolo_bricks.py

* fix

* Update yolo_bricks.py
2022-11-20 09:13:35 +08:00
Range King cb93fb6754
fix links in config.md (#303) 2022-11-20 09:12:49 +08:00
HinGwenWoong 90494b44f2
[DOC] Add YOLOv6 description doc - overview (#252)
* [DOC] Add YOLOv6 description doc

* [DOC] Add YOLOv6 description doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc
2022-11-19 10:22:26 +08:00
fatih 0fd6444c00
[Feature] Add large image demo with `sahi` (#284)
* add large image demo with sahi

* fix some typos

* restructure based on reviews

* update default patch size

* add docstring and update docs

* updates based on reviews

* print information

* add debug, update docs, add large image sample

* update docs

* update docs

* update docs

* direct user to install sahi
2022-11-18 20:30:43 +08:00
jason_w 5cee9c977d
fix `make_stage_plugins` doc in basebackbone (#296)
* fix 'make_stage_plugins' doc

* Update the parameter name to be consistent with the document
2022-11-18 20:25:36 +08:00
Range King 1c8e581d2c
[Docs] Add two examples of backbone replacement and update `plugin.md` in EN (#291)
* Update how_to.md

* delete toc

* Update how_to.md

* Update plugin.md

* Update how_to.md

* Update how_to.md

* replace todolist

* fix pretrained model link

* Update how_to.md

* refine configs
2022-11-18 09:55:50 +08:00