Commit Graph

272 Commits (e62c8c4593702d23e5a75d940d9158aa4ea418ae)
 

Author SHA1 Message Date
Haian Huang(深度眸) e62c8c4593
Bump version to 0.4.0
Bump version to 0.4.0
2023-01-18 19:54:21 +08:00
Haian Huang(深度眸) d8b13538dc
Add v0.4.0 Changelog (#477)
* Add v0.4.0 changelog

* update

* fix

* fix

* fix

* update

* update

* add url
2023-01-18 19:43:27 +08:00
Range King ac27f900e3
[Docs] Refine README (#482)
* add emoji in README

* update

* update

* update1

* update1

* Add banner

* minor fix

* adjust position

* add toc

* update zh_CN

* update zh_CN

* test

* update

* test

* test

* update

* update

* update

* update

* add banner in README_zh-CN

* replace emoji

* minor fix

* remove some videos

* Update README_zh-CN.md

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-18 19:38:34 +08:00
Range King 51d30a2837
[Docs] Update `article.md` and `model_zoo.md` (#481)
* Update article.md

* Update model_zoo.md

* Update model_zoo.md
2023-01-18 15:23:31 +08:00
Range King 6bdfc0da26
[Feature] Add a tool for scheduler visualization (#479)
* Create vis_scheduler.py

* add docs

* Update .gitignore

* Update useful_tools.md

* add source link

* Add TODO
2023-01-18 15:03:02 +08:00
Haian Huang(深度眸) 6f478e1ae0
[Bug] Support RTMDet P6 model (#480) 2023-01-18 15:02:39 +08:00
Youfu da62ae7283
[Doc] Add PPYOLOE model structure (#471)
* Update README.md

Add PPYOLOE model structure

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md
2023-01-18 14:32:20 +08:00
Nioolek 07ab35222c
[Feature] Support YOLOv8 training (#458)
* add v8 training head

* add training config

* update

* update1

* update all config

* update all config

* update metafile

* add head loss

* update readme

* iou2ciou in tal, add config

* change init layer

* update

* update

* add clip_grad

* update max_aspect_ratio

* add init

* fix aug config

* update forward

* add official v8 loss

* add official v8 loss

* 同步预测方式

* fix loss bug

* delete useless code

* 替换iou,但还有问题

* fix bug

* fix empty

* add UT

* add UT

* update readme

* replace iou_cal

* add use_ciou docstr

* update metafile

* update metafile

* update metafile

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-18 14:16:37 +08:00
LinXiaoZheng 8eefed39ef
[Doc] Fix the installation step on get_started.md (#474) 2023-01-17 15:15:24 +08:00
Haian Huang(深度眸) c32ca4549f
[DOC] Add YOLOv8 docs (#461)
* Add YOLOv8 docs

* update

* update

* fix some error

* update en

* fix comment

* update
2023-01-17 15:12:02 +08:00
tripleMu feef44da23
[Fix] Fix requirement about only ONNXRuntime without TensorRT (#469)
* Fix requirement about only onnxruntime without tensorrt

* Update projects/easydeploy/model/backendwrapper.py

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

* Update projects/easydeploy/model/backendwrapper.py

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

* Update projects/easydeploy/model/backendwrapper.py

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

* Update projects/easydeploy/model/backendwrapper.py

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

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-16 19:36:49 +08:00
Range King e66991fc6e
[Fix] Fix YOLOv8 graph and a lint issue (#459)
* add missing bg frame

* fix easydeploy lint
2023-01-12 16:00:46 +08:00
tripleMu 8e39252259
Support YOLOv8 deploy (#456) 2023-01-12 10:07:44 +08:00
Range King 550664d514
[Doc] Add the model structure graph of YOLOv8-P5 (#455)
* add YOLOv8 model structure figure

* Fix the layer name
2023-01-12 10:07:06 +08:00
HinGwenWoong 935d710f79
[Feature] Support YOLOv8 inference (#445)
* Add backbone

* Improve code

* fix

* Add neck

* Add config

* Fix layer param

* Fix layer param

* Add head

* Add head

* Add model converter

* Add model converter

* Add model converter

* Improve code

* update

* update

* align test

* Improve code

* Improve code

* Improve code

* Fix lint

* Fix lint

* Fix lint

* Fix lint

* Improve code

* Improve code

* Improve code

* Add configs

* Improve doc

* Improve doc

* update

* Fix config

* update

* Fix config

* Fix config

* update

* Fix config epoch

* update

* Fix docstr

* Add UT

* Add UT

* Fix doc

* update

* Fix lint

* Fix doc

* Fix config name

* Improve config

* Improve default

* Add docstr

* Improve code

* Improve code

* Drop `bbox_head.head_module.dfl.conv.weight` when convert to mmyolo weight

* Delete useless code

* Add batch_shapes_cfg but not enable it.

* update

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-11 19:01:06 +08:00
tripleMu f1855ca618
[Feature] Add new YOLOX bbox coder for TensorRT-8 and fix TensorRT-7 bug (#450)
* Add yolox bbox decoder for tensorrt backend

* Fix tensorrt7+ bug
2023-01-11 17:10:04 +08:00
Yijie Zheng fd0f07b8ac
[Fix] Fix import error in `optimize_anchors.py` (#452)
* [Fix] Fix bugs in the output annotation file when  is set.

* Update labelme2coco.py

* Update labelme2coco.py

* Improve code

* [Fix]Fix import error in optimize_anchors.py

* [Fix]Fix import error in optimize_anchors.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-11 17:09:31 +08:00
tianlei c682ae82c6
adjust the minimum supported python version to 3.9 (#449) 2023-01-09 18:31:37 +08:00
Haian Huang(深度眸) b0cf073241
Bump version to v0.3.0
Bump version to v0.3.0
2023-01-08 15:48:53 +08:00
Haian Huang(深度眸) 9c6b4fedc8
Add v0.3.0 changelog (#440)
* bump version to v0.3.0

* add v7 cat demo config

* update mmdet version

* update

* fix topk out of range error

* update

* add changelog.md

* update

* update

* update

* update

* update
2023-01-08 15:31:58 +08:00
Haonan fc107b8997
[Docs] Add an example project to 'projects\' folder (#401)
* add_example_project

* renew readme

* Update projects/example_project/README.md

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

* Update projects/example_project/README.md

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

* Update projects/example_project/README.md

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

* Update projects/example_project/README.md

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

* Update projects/example_project/README.md

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

* Update projects/example_project/README.md

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

* Update projects/example_project/dummy/dummy_resnet.py

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

* Update projects/example_project/README.md

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

* Update projects/example_project/dummy/dummy_resnet.py

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

* Update projects/example_project/README.md

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

* renew to yolov5

* Update projects/example_project/dummy/dummy_yolov5cspdarknet.py

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

* Update projects/example_project/README.md

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

* Update projects/example_project/dummy/dummy_yolov5cspdarknet.py

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

* fix issue

* Update projects/example_project/dummy/dummy_yolov5cspdarknet.py

* Update projects/example_project/README.md

* Update projects/example_project/README.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2023-01-08 14:51:39 +08:00
Range King a631370cb3
Update model graphs (#442) 2023-01-07 17:44:19 +08:00
Seperendity 89275adb29
[CodeCamp] #50 Add module combination doc (#349)
* the file  to verify the installation

* "add module_combination"

* add module_combination

* modified module_combination.md

* modify module_combination.md

* fix module_combination.md
2023-01-06 20:30:48 +08:00
Jing Tang 7d07e1a780
Update yolo_detector.py (#441)
将 YOLOX -> YOLO
2023-01-06 19:20:24 +08:00
Nioolek 07afc3ee78
Fix potential bugs in rtmdet multi-scale training (#439)
* Distinguish variables when training and testing

* beauty
2023-01-06 19:03:14 +08:00
Haian Huang(深度眸) 884330108d
[BC Breaking] Keep the same input shape order as mmdet 3.0.0rc5 (#436)
* Keep the same input shape order as mmdet

* add desc

* update

* update
2023-01-06 19:00:30 +08:00
Nioolek 8127805dd3
[Feature] Support PPYOLOE training (#259)
* add ppyoloe backbone, neck

* add ppyoloe test

* add docstring

* add ppyoloe m/l/x configfile

* add ppyoloe_coco.py

* rename config

* add typehint

* format code; add ut

* add datapre

* add datapre

* add ppyoloe datapre

* add ppyoloe datapre

* add ppyoloe datapre

* reproduce coco v0.1

* add ut

* add ut, docstring

* fix transforms bug

* use mmdet dfloss

* add non plus model config

* add non plus model config

* fix

* add ut

* produce coco v0.2

* fix config

* fix config

* fix eps and transforms bug

* add ema

* fix resize

* fix transforms.py

* fix transforms.py

* fix transforms.py

* old version

* old version

* old version

* old version

* old version

* old version

* fix stride loss error

* add INTER_LANCZOS4

* fix crop bug

* init commit

* format code

* format code

* bgr transforms.py

* add typehint and doc in transforms.py

* 继承新版yolov6head写法,删除不必要的注释

* fix transforms var name bug

* bbox decode use stridetensor insted of priors

* add headmodule todo

* add ppyoloe README.md

* add ppyoloe README.md

* Update tests/test_datasets/test_transforms/test_transforms.py

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

* Update tests/test_datasets/test_transforms/test_transforms.py

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

* save ckpt last 10 epochs

* save_best ckpt

* del ppyoloe collate

* change name of ppyoloebatchrandomresize

* add the reason for rewritten PPYOLOEDetDataPreprocessor

* solve ppyoloerandomresize name error

* rm PPYOLOERandomExpand

* rm l1 loss

* rm yolov6 loss_obj

* add difference between yolov6 and ppyoloe

* add reason for rewrite paramscheduler

* change proj init way

* fix error

* rm proj_conv in pth

* format code

* add load_from

* update

* support fast training

* add pretrained model url

* update

* add pretrained model url

* fix error

* add imagenet model convert and use init_cfg to init backbone

* add plus model pretrain model

* add ut

* add ut

* fix ut

* fix withstride bug

* cat in yolov5_collate

* merge

* fix typehint

* update readme

* add reason for gap

* fix log in README.md

* rollback yolov6

* change inherit

* fix ut

* fix ut

Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: hha <1286304229@qq.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-06 15:54:39 +08:00
wzr-skn a20b160f0f
[CodeCamp]Add doc of yolov6_description.md (#382)
* YOLOv6网络结构文档

* 修正YOLOv6网络结构文档

* 修改模型结构图四周多余的留白

Co-authored-by: yl4203 <wsnbbzwd19960817>
2023-01-06 14:59:34 +08:00
MingJian.L c14ecc336c
[DOC] Update useful_tools.md (#384)
* Update useful_tools.md

更新了浏览数据部分的文档

* Update useful_tools.md

* Update useful_tools.md

* Update useful_tools.md

* Update docs/zh_cn/user_guides/useful_tools.md

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

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-06 14:48:17 +08:00
tang576225574 2f399df2d6
[Feature] Add the unit test to test the `scale_factor` in LetterResize (#408)
* correct the scale_factor and pad_param value in LetterResize

* correct the scale_factor and pad_param value in LetterResize

* correct the scale_factor and pad_param value in LetterResize

* tang

* update

* tang

* tang

* fix linting

* tang

* tang

* tang

Co-authored-by: hha <1286304229@qq.com>
2023-01-06 14:40:21 +08:00
Yijie Zheng 2290713ae6
[Fix] Fix bugs in the output annotation file when `--class-id-txt` is set. (#430)
* [Fix] Fix bugs in the output annotation file when  is set.

* Update labelme2coco.py

* Update labelme2coco.py

* Improve code

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-06 11:41:17 +08:00
jason_w d8cbf216de
[Fix] Fix typo in docstring (#433)
* fix typo in docstring

* fix typo in docs

* fix typo in docs
2023-01-06 11:40:47 +08:00
RangiLyu 5172805dae
fix rtmdet-s cfg and neck (#434) 2023-01-06 11:40:19 +08:00
tripleMu 28439f102a
[Deploy] Support YOLOv7 easy deploy (#427)
* Support YOLOv7 easy deploy

* Remove useless comment

* Fix mkdir
2023-01-04 20:10:58 +08:00
tripleMu 7c5e21c39f
Fix customize dataset training (#428) 2023-01-04 20:02:02 +08:00
RangiLyu 48f8896e84
[Feature] Implement fast version of RTMDet. (#425)
* Accelerate RTMDet

* update

* update

* update

* update1

* update2

* update pipeline

* update lr cudnnbenchmark

* revert batchsize

* fix batch inference

* refactor head

* update box

* bs=16

* update

* move reduce mean

* update head

* per img loss

* fix

* fix sum

* concat loss

* batch dsla

* sort topk

* bs 32

* clean code

* update readme

* update ut

* update checkpoint

* num_class

* clean code

* resolve comments

* fix readme

* fix ut

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-04 16:06:44 +08:00
Xin Li c7a9026812
url fix in README (#426) 2023-01-03 15:43:23 +08:00
kitecats 14d9cab354
[CodeCamp] Add docs about how to freeze the weight of backbone or neck (#418)
* Add docs about how to freeze the weight of backbone or neck

* refine the docs

* refine the docs

* refine the docs
2022-12-31 16:36:40 +08:00
jason_w b1f7e80fe3
Fix typehint in Heads (#416) 2022-12-31 16:35:18 +08:00
jason_w 13de22dfd2
Fix typehint in YOLOv6 Head (#415) 2022-12-29 21:21:44 +08:00
RangiLyu cdc359c2de
fix batch inference bug (#413) 2022-12-29 17:18:54 +08:00
HinGwenWoong 99af3c980f
[Fix] Set metainfo fields to lower case and fix CI (#412)
* Set metainfo fields to lower case

* Set metainfo fields to lower case

* Fix ci

* Fix ci
2022-12-29 16:23:52 +08:00
tripleMu a44495868d
[Feature] Support PyTorch model forward for TensorRT inference (#377)
* Support pytorch model forward for TensorRT

* Add ort wrapper

* Fix import

* Add deploy image-demo

* raise NotImplementedError

* Fix onnxruntime output gpu tensor
2022-12-29 12:11:16 +08:00
tianlei 66c80e91e1
[Doc] update the English version of `custom_dataset.md` (#381)
* part_1

* [WIP][Doc]update the part of 1. Prepare dataset

* update to 5. create a new config file

* update to 9.2 perform training

* add pre-commit

* 下一次更新内容

* update to 11.1.2 Creating a Docker container

* add pre-commit

* all updates complete!

* Update docs/en/user_guides/custom_dataset.md

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

* Update docs/en/user_guides/custom_dataset.md

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

* Update docs/en/user_guides/custom_dataset.md

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

* modify en

* Update docs/en/user_guides/custom_dataset.md

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

* Update docs/en/user_guides/custom_dataset.md

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

* sync:Refactor dataset metainfo to lowercase

* Update docs/en/user_guides/custom_dataset.md

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

* Update docs/en/user_guides/custom_dataset.md

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

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-12-28 20:22:10 +08:00
kitecats d0a8aaa284
[CodeCamp] Add don't used pre-training weights to how_to (#404)
* Add Don't used pre-training weights to how_to

* refine docs of how_to

* refine doc of how_to

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

* refine the doc

* refine the doc

Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-12-28 17:00:29 +08:00
Nioolek 9ef883187a
[Feature] Show YOLOv5 assigner results (#383)
* init commit

* init commit

* init commit

* 定稿,开始重构

* format code

* format code

* add typehint and doc

* init commit

* init commit

* init commit

* 定稿,开始重构

* format code

* format code

* add typehint and doc

* format code

* rollback

* add doc

* fix less img bug

* format code

* format code

* add README.md

* beauty

* beauty

* uniform name

* uniform name

* uniform name

* uniform name
2022-12-28 16:40:14 +08:00
Range King bb4aea90da
[Refactor] Refactor dataset metainfo to lowercase (#362)
* [Refactor] Refactor dataset metainfo to lowercase.

* Add docs

* Refactor metainfo to lowercase in new files

* Fix dataset meta compatibility
2022-12-27 21:19:39 +08:00
tripleMu 81b7942de1
[Feature] Support `iscrowd` attribute training in YOLOv5 (#260)
* [Feature] Support YOLOv7 inference (#149)

* update

* update

* update

* update

* update

* add docstr

* fix comments

* update

* [Feature] support mmyolo deployment (#79)

* support mmyolo deployment

* mv deploy place

* remove unused configs

* add deploy code

* fix new register

* fix comments

* fix dependent codebase register

* remove unused initialize

* refact deploy config

* credit return to triplemu

* Add yolov5 head rewrite

* refactor deploy

* refactor deploy

* Add yolov5 head rewrite

* fix configs

* refact config

* fix comment

* sync name after mmdeploy 1088

* fix mmyolo

* fix yapf

* fix deploy config

* try to fix flake8 importlib-metadata

* add mmyolo models ut

* add deploy uts

* add deploy uts

* fix trt dynamic error

* fix multi-batch for dynamic batch value

* fix mode

* fix lint

* sync model.py

* add ci for deploy test

* fix ci

* fix ci

* fix ci

* extract script to command for fixing CI

* fix cmake for CI

* sudo ln

* move ort position

* remove unused sdk compile

* cd mmdeploy

* simplify build

* add missing make

* change order

* add -v

* add setuptools

* get locate

* get locate

* upgrade torch

* change torchvision  version

* fix config

* fix ci

* fix ci

* fix lint

Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>

* [Feature] Support YOLOv5 YOLOv6  YOLOX Deploy in mmdeploy (#199)

* Support YOLOv5 YOLOv6 YOLOX Deploy in mmdeploy

* Fix lint

* Rename _class to detector_type

* Add some common

* fix lint

Co-authored-by: huanghaian <huanghaian@sensetime.com>

* [DOC] Add deploy guide doc (#220)

* Add deploy doc-0

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Update 部署必备指南.md

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

* Del unused line

* Add docs

* Fix

* Fix

* Rename docs

* Rename docs

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

* [Feautre] Add deploy dockerfile (#224)

* Add dockerfile for deploy

* Fix

* Update docker/Dockerfile_deployment

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

* Update docker/Dockerfile_deployment

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

* Add opencv old version

* Add dockerfile and fix some typo

* Remove repeat packages

* Fix undefined symbol bug

* Update docker/Dockerfile_deployment

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

* Update docker/Dockerfile_deployment

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

* Add docs for deploy dockerfile

* Fix typo

* Update docs/zh_cn/advanced_guides/yolov5_deploy.md

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

* Add profiler

* Update docs/zh_cn/advanced_guides/yolov5_deploy.md

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

* Update docker/Dockerfile_deployment

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

* Remove mmcv-full

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Add changelog of v0.1.2 (#226)

* Add changelog of v0.1.2

* update version

* fix comments

* fix comments

* update

* update version

* update version

* Support ignore

Cal loss with ignore

Typo

Fix import from mmdet

Fix train_cfg is None

* Fix judgement

* Fix bug and add a demo config

* Add new config

* remove

* rm

* remove loadanno

* Try 0

* Try-2

* fix

* add avg_factor

* update

* support rectangle training

* add ignore cfg

* update

* delete

* Fix

* Add unitest

* Add ignore_iof_thr

* Add test

Co-authored-by: Haian Huang(深度眸) <huanghaian@pjlab.org.cn>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-12-27 14:18:10 +08:00
caj-github 929b403fb0
Fix RuntimeError of `torch.cat()` expected a non-empty list of Tensors (#376)
* RuntimeError: torch.cat(): expected a non-empty list of Tensors

* blank

* fix lint

* update

Co-authored-by: hha <1286304229@qq.com>
2022-12-27 12:01:10 +08:00
zhubochao 6c5acd238c
Fix the device inconsistency error in yolov7 training (#397)
* [FIX] Indices should be either on cpu or on the same device as the indexed tensor

* fix lint error

* use new_full instead of simple torch.ones

* fix lint error

* fix type error

* use existed tensor to create new_full tensor

Co-authored-by: zhubochao <zhubochao@zhubochaodeMacBook-Pro.local>
2022-12-27 11:54:28 +08:00