Commit Graph

11 Commits (main)

Author SHA1 Message Date
Damian Deza 1cb8aa5c9f
Fixed issue with confidence for single class detectors when exporting ()
* Fixed issue with confidence for single class detectors when exporting

* Typo
2022-09-12 17:28:19 +03:00
tripleMu 8dc755a359
Add grid concat and fuse such operators ()
* 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
Taka 8758cdd838
fix darwin platform string () 2022-08-08 10:17:04 +03:00
Taka cfde076fb4
Fix coreml quantization ()
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
AlexeyAB84 c51c13a23f Fixed CoreML export 2022-08-02 01:40:08 +03:00
Ian 0563c70705
Add TorchScript-Lite export ()
* TorchScript-Lite export

* forgot the import
2022-08-02 00:46:25 +03:00
tripleMu a7c00297d5
Support dynamic batch for TensorRT and onnxruntime ()
* 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
tripleMu 1c59e43d9d
Add end2end yolov7 onnx export for TensorRT8.0+ and onnxruntime(testing now) ()
* 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
AlexeyAB84 e5928b095a minor fix 2022-07-22 17:04:38 +03:00
Linaom1214 afdc86f519
End2end ()
* export end2end onnx model

* fixbug

* add web demo ()

* 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 ()

* fuse IDetect ()

* Fixes  ()

* 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
linghu8812 96390ed201
support onnx to tensorrt convert () 2022-07-21 20:06:24 +03:00