Commit Graph

114 Commits (mask)
 

Author SHA1 Message Date
Kin-Yiu, Wong 4ebeeca405
mask code
fix https://github.com/WongKinYiu/yolov7/issues/855
2022-09-28 16:58:35 +08:00
Kin-Yiu, Wong fb797838bb
mask code
update mt
2022-08-18 12:55:25 +08:00
Kin-Yiu, Wong 892dfbce7d
mask code
update refine
2022-08-17 12:48:17 +08:00
Kin-Yiu, Wong b0211315f5
Update README.md 2022-08-10 11:34:28 +08:00
Kin-Yiu, Wong f2fe402808
Add files via upload 2022-08-10 11:27:32 +08:00
Kin-Yiu, Wong af105684aa
Create yolov7-mask.yaml 2022-08-10 11:26:23 +08:00
Kin-Yiu, Wong be1109c92a
Update general.py 2022-08-10 11:25:30 +08:00
Kin-Yiu, Wong a3c802d72e
Create hyp.scratch.mask.yaml 2022-08-10 11:23:31 +08:00
Kin-Yiu, Wong 9764197bd2
Update common.py 2022-08-10 11:22:12 +08:00
Kin-Yiu, Wong 1b0db446cf
Update yolo.py 2022-08-10 11:21:21 +08:00
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