Commit Graph

97 Commits (c14ba0c297b3b5fc0374c917db798c88f9dd226c)
 

Author SHA1 Message Date
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
linghu8812 96390ed201
support onnx to tensorrt convert (#114) 2022-07-21 20:06:24 +03:00
Kin-Yiu, Wong 4f6e390c99
main code
fix nan of aux training https://github.com/WongKinYiu/yolov7/issues/250#issue-1312356380 @hudingding
2022-07-21 12:09:25 +08:00
Kin-Yiu, Wong de6a5e733d
main code
fix ciou
2022-07-21 11:48:57 +08:00
Alexey 8730e4ba4f
Update README.md 2022-07-21 04:53:38 +03:00
Kin-Yiu, Wong 1e1e0d52d9
main code
update trace
2022-07-20 14:46:12 +08:00
Kin-Yiu, Wong 8b72ac7792
main code
update pose
2022-07-18 13:45:26 +08:00
Kin-Yiu, Wong 2e86053662
main code
update pose
2022-07-18 13:37:31 +08:00