Commit Graph

104 Commits (3708cdedd2d9348d2c2c0588f62bb1fe574cd1ec)
 

Author SHA1 Message Date
Kin-Yiu, Wong 3708cdedd2
main code
update anchor https://github.com/WongKinYiu/yolov7/issues/272
2022-08-09 20:06:36 +08:00
Kin-Yiu, Wong 54d3908a0c
main code
update anchors
2022-08-09 15:23:42 +08:00
Wang Xin fef71ef42c
update gitignore (#461) 2022-08-09 09:28:28 +03:00
AlexeyAB84 892b603718 Added .gitignore 2022-08-09 08:07:05 +03:00
AlexeyAB84 711a16ba57 Added param loss_ota for hyp.yaml, to disable OTA for faster training 2022-08-09 07:48:28 +03:00
AlexeyAB84 469a4d0e8d minor fix 2022-08-09 07:46:47 +03:00
tripleMu 8dc755a359
Add grid concat and fuse such operators (#389)
* Add grid concat and fuse so many op

* Fix model

* Fix other detector

* Update yolo.py

* Update yolo.py

Co-authored-by: Alexey <AlexeyAB@users.noreply.github.com>
2022-08-09 07:26:05 +03:00
spacewalk01 c14ba0c297
fixed the problem of downloading models (#451) 2022-08-08 10:17:40 +03:00
Taka 8758cdd838
fix darwin platform string (#454) 2022-08-08 10:17:04 +03:00
Kayce001 4789bd1a4f
Update keypoint.ipynb (#292)
* Update keypoint.ipynb

add "with torch.no_grad():" in line 60.

* Update keypoint.ipynb
2022-08-06 06:39:28 +03:00
AlexeyAB84 591c5b0236 minor fix 2022-08-06 01:45:25 +03:00
Taka cfde076fb4
Fix coreml quantization (#434)
Instead of importing platform, you can check it by `sys.platform`. By the way, `prefix` is a missing value in this script.
2022-08-06 00:13:32 +03:00
Kadir Nar 1560034c77
added cpu parameter support (#423) 2022-08-06 00:10:21 +03:00
Kin-Yiu, Wong 72a9c9628a
main code
update https://github.com/WongKinYiu/yolov7/issues/272
2022-08-04 20:20:51 +08:00
Mohammad Khoshbin b8956dd5a5
fix training with frozen layers (#378) 2022-08-02 17:55:28 +03:00
Linaom1214 1e51f564e0
update readme (#396) 2022-08-02 17:55:01 +03:00
AlexeyAB84 c51c13a23f Fixed CoreML export 2022-08-02 01:40:08 +03:00
Ian 0563c70705
Add TorchScript-Lite export (#387)
* TorchScript-Lite export

* forgot the import
2022-08-02 00:46:25 +03:00
AlexeyAB84 0a58f6d1cb minor fix 2022-08-02 00:45:17 +03:00
Mohammad Khoshbin cc42a206ef
fix cpu-only training CUDA error (#369) 2022-07-30 18:43:18 +03:00
tripleMu 6bacefff5c
Unified output names (#283) 2022-07-29 04:58:57 +03:00
AlexeyAB84 5f7d38b12a minor fix 2022-07-29 01:38:49 +03:00
philipp-schmidt 8eee99fcc5
Add Triton Inference Server deployment (#346)
* Add client code

* Add README.md

Co-authored-by: Philipp Schmidt <philipp.schmidt@isarsoft.com>
2022-07-29 01:01:59 +03:00
tripleMu a7c00297d5
Support dynamic batch for TensorRT and onnxruntime (#329)
* Support dynamic batch for TensorRT and onnxruntime

* Fix output name

* Add some images

* Add dynamic-batch usage notebook

* Add example notebook for onnxruntime and tensorrt
2022-07-29 01:01:43 +03:00
Dhia_Oussayed 0d882e553e
Fixed a bug for Hyperparameters Evolution. (#344)
* Update hyp.scratch.custom.yaml

anchors parameter missing, evolution won't start without.

* Update train.py

Updated the hyperparameter evolution metadata variable to match the hyp.yaml files for the evolution to run successfully.

* Update hyp.scratch.p5.yaml

Added the anchors parameter, evolution don't start without it.

* Update hyp.scratch.p6.yaml

added the anchors parameter for the hyperparameter evolution.

* Update hyp.scratch.tiny.yaml

added the anchors parameter for the hyperparameters evolution

* Update hyp.scratch.custom.yaml

* Update hyp.scratch.tiny.yaml

* Update hyp.scratch.p5.yaml

* Update hyp.scratch.p6.yaml

* Update train.py

* Update train_aux.py
2022-07-28 22:09:52 +03:00
AlexeyAB84 264fc09886 Merge branch 'main' of https://github.com/WongKinYiu/yolov7 2022-07-28 22:03:15 +03:00
AlexeyAB84 954cde65ab Fuse IAuxDetect 2022-07-28 22:03:09 +03:00
Mohammad Khoshbin 2a731fc5d2
fix reparametrization for any nc (#342) 2022-07-28 18:49:06 +03:00
AlexeyAB84 4c207e1ae6 minor fix 2022-07-28 02:04:34 +03:00
Mohammad Khoshbin 13594cf6d4
add tensor.detach for yolo_pose bug (#333) 2022-07-28 01:35:17 +03:00
ksnzh 86cc5d0c4f
fix IDetect fuseforrward onnx export (#332) 2022-07-27 19:25:06 +03:00
Mohammad Khoshbin a4724d5592
non-max-suppression function fixed for one class model conf. (#305) 2022-07-27 17:47:28 +03:00
tripleMu 7a7cba744a
Fix yolo for torch.onnx.export (#314) 2022-07-26 18:03:20 +03:00
Kin-Yiu, Wong b3c41653ba
main code
upload paper pdf
2022-07-25 15:56:41 +08:00
AlexeyAB84 da5fe65e90 minor fix 2022-07-25 05:13:04 +03:00
Alexey d10b1835c7
Update README.md 2022-07-25 05:12:40 +03:00
AlexeyAB84 6d53a210ad Merge branch 'main' of https://github.com/WongKinYiu/yolov7 2022-07-25 04:27:35 +03:00
AlexeyAB84 703c2ff21a minor fix 2022-07-25 04:27:28 +03:00
Alexey d489e9435b
Update README.md 2022-07-25 04:01:02 +03:00
AlexeyAB84 82f26becad Added YOLOv7onnx.ipynb and YOLOv7trt.ipynb 2022-07-25 03:52:20 +03:00
Kin-Yiu, Wong 13458cd4a2
main code
update readme
2022-07-24 22:26:53 +08:00
Alexey 26c26ce234
Update README.md 2022-07-23 05:45:07 +03:00
tripleMu 1c59e43d9d
Add end2end yolov7 onnx export for TensorRT8.0+ and onnxruntime(testing now) (#273)
* Add end2end yolov7 onnx export for TensorRT8.0+

* Add usage in README

* Update yolo.py

* Update yolo.py

* Add tensorrt onnxruntime examples

* Add usage in README

Co-authored-by: Alexey <AlexeyAB@users.noreply.github.com>
2022-07-23 05:36:51 +03:00
Alexey aae70703f7
Update add_nms.py 2022-07-23 04:54:18 +03:00
AlexeyAB84 a5b853c3fa minor fix 2022-07-22 17:25:11 +03:00
AlexeyAB84 e5928b095a minor fix 2022-07-22 17:04:38 +03:00
Linaom1214 afdc86f519
End2end (#61)
* export end2end onnx model

* fixbug

* add web demo (#58)

* Update README.md

* main code

update yolov7-tiny deploy cfg

* main code

update yolov7-tiny training cfg

* main code

@liguagua752109150 https://github.com/WongKinYiu/yolov7/issues/33#issuecomment-1178669212

* main code

@albertfaromatics https://github.com/WongKinYiu/yolov7/issues/35#issuecomment-1178800685

* main code

update link

* main code

add custom hyp

* main code

update default activation function

* main code

update path

* main figure

add more tasks

* main code

update readme

* main code

update reparameterization

* Update README.md

* main code

update readme

* main code

update aux training

* main code

update aux training

* main code

update aux training

* main figure

update yolov7 prediction

* main code

update readme

* main code

rename

* main code

rename

* main code

rename

* main code

rename

* main code

update readme

* main code

update visualization

* main code

fix gain for train_aux

* main code

update loss

* main code

update instance segmentation demo

* main code

update keypoint detection demo

* main code

update pose demo

* main code

update pose

* main code

update pose

* main code

update pose

* main code

update pose

* main code

update trace

* Update README.md

* main code

fix ciou

* main code

fix nan of aux training https://github.com/WongKinYiu/yolov7/issues/250#issue-1312356380 @hudingding

* support onnx to tensorrt convert (#114)

* fuse IDetect (#148)

* Fixes #199 (#203)

* minor fix

* resolve conflict

* resolve conflict

* resolve conflict

* resolve conflict

* resolve conflict

* resolve

* resolve

* resolve

* resolve

Co-authored-by: AK391 <81195143+AK391@users.noreply.github.com>
Co-authored-by: Alexey <AlexeyAB@users.noreply.github.com>
Co-authored-by: Kin-Yiu, Wong <102582011@cc.ncu.edu.tw>
Co-authored-by: linghu8812 <36389436+linghu8812@users.noreply.github.com>
Co-authored-by: Alexander <84590713+SashaAlderson@users.noreply.github.com>
Co-authored-by: Ben Raymond <ben@theraymonds.org>
Co-authored-by: AlexeyAB84 <alexeyab84@gmail.com>
2022-07-22 16:24:13 +03:00
AlexeyAB84 2596994f39 minor fix 2022-07-21 20:37:15 +03:00
Ben Raymond 894a93e318
Fixes #199 (#203) 2022-07-21 20:07:49 +03:00
Alexander dc3e04087d
fuse IDetect (#148) 2022-07-21 20:07:08 +03:00