yolov5_ZWgra

Ankle/yolov5_ZWgra

0 Fork 0

Commit Graph

Select branches

Hide Pull Requests

G0_raw_decode_output

G0_yolov5s6_raw

G10_decoupled_head_light_FGEM_safe

G11_cls_head_enhance_Lightweight

G12_cls_head_enhance_and_ECA

G13_decoupled_head_ECA

G1_backbone

G2_neck

G2_neck_2

G3_head

G4_cls_loss_gain

G5_focal_loss

G6_decoupled_head

G7_RawHead_add_ECA

G8_decoupled_head_light

G9_decoupled_head_light_FGEM

M0_yolov5_raw

master

c84dd27d62 New val.py cuda variable (#6957) Glenn Jocher 2022-03-12 12:57:08 +01:00
c6b4f84fd1 Update Dockerfile torch==1.11.0+cu113 (#6954) Glenn Jocher 2022-03-12 00:45:07 +01:00
b94b59e199 DetectMultiBackend() --half handling (#6945) Glenn Jocher 2022-03-11 16:31:52 +01:00
84efa62b2d Fix PyTorch Hub export inference shapes (#6949) Glenn Jocher 2022-03-11 16:18:40 +01:00
caf7ad0500 Allow 3-point segments (#6938) Glenn Jocher 2022-03-10 18:41:47 +01:00
055e72af5b Optimize PyTorch 1.11.0 compatibility update (#6933) Glenn Jocher 2022-03-10 12:58:41 +01:00
d3d9cbce22 PyTorch 1.11.0 compatibility updates (#6932) Glenn Jocher 2022-03-10 12:41:06 +01:00
6dd82c0252 Move git_describe() to general.py (#6918) Glenn Jocher 2022-03-09 18:22:53 +01:00
e6e36aac10 Update bytes to GB with bitshift (#6886) Glenn Jocher 2022-03-07 19:26:37 +01:00
acc58c1dcf Fix TRT max_workspace_size deprecation notice (#6856) Glenn Jocher 2022-03-07 13:52:53 +01:00
a5a1760ea6 Bump actions/checkout from 2 to 3 (#6881) dependabot[bot] 2022-03-07 13:49:27 +01:00
c8a589920e Bump actions/setup-python from 2 to 3 (#6880) dependabot[bot] 2022-03-07 13:48:59 +01:00
596de6d5a0 Default FP16 TensorRT export (#6798) DavidBandGlenn Jocher 2022-03-07 03:21:16 +07:00
7e98b4801a Update yolov5s.yaml (#6865) vnekatandGlenn Jocher 2022-03-07 00:50:01 +05:30
4728840745 Update --cache disk deprecate *_npy/ dirs (#6876) Glenn Jocher 2022-03-06 16:16:17 +01:00
8a66ebad44 Add *.ts to VID_FORMATS (#6859) Glenn Jocher 2022-03-04 14:10:13 +01:00
601dbb83f0 AutoAnchor improved initialization robustness (#6854) Glenn Jocherandpre-commit-ci[bot] 2022-03-04 10:32:18 +01:00
bcc92e2169 Update sweep.yaml (#6825) Louis Combaldieu 2022-03-04 09:39:23 +01:00
63ddb6f0d0 Update autoanchor.py (#6794) Glenn Jocher 2022-02-26 19:15:12 +01:00
c2403eb04c Update PULL_REQUEST_TEMPLATE.md (#6783) Glenn Jocher 2022-02-25 14:01:58 +01:00
9ec51a6396 YOLOv5s6 params FLOPs fix (#6782) Glenn Jocher 2022-02-25 13:59:21 +01:00
0f819919ad Update EMA decay tau (#6769) Glenn Jocher 2022-02-25 12:33:09 +01:00
b2adc7c39a Fix export for 1-channel images (#6780) Louis Combaldieu 2022-02-25 10:56:37 +01:00
cea994b3f6 Update hyp.VOC.yaml (#6772) Glenn Jocher 2022-02-25 00:14:17 +01:00
741cd0eb23 Update tutorial.ipynb (#6771) Glenn Jocher 2022-02-25 00:12:27 +01:00
c161557563 Default OMP_NUM_THREADS=8 (#6770) Glenn Jocher 2022-02-25 00:09:14 +01:00
47f265dde2 Update min warmup iterations from 1k to 100 (#6768) Glenn Jocher 2022-02-24 23:37:25 +01:00
4bab56526b Update tutorial.ipynb (2 CPUs, 12.7 GB RAM, 42.2/166.8 GB disk) (#6767) Glenn Jocher 2022-02-24 18:29:59 +01:00
66aaf5184d Pre-commit table fix (#6744) Glenn Jocher 2022-02-22 19:35:31 +01:00
3752807c0b YOLOv5 v6.1 release (#6739) Glenn Jocher 2022-02-22 12:35:24 +01:00
2692e67c5f Updated VOC hyperparameters (#6732) Glenn Jocher 2022-02-22 12:11:18 +01:00
a936f5f219 Switch default LR scheduler from cos to linear (#6729) Glenn Jocher 2022-02-21 14:23:27 +01:00
dbbb57cf0b GH: add PR template (#6482)
Jirka Borovec Glenn Jocher pre-commit-ci[bot]
2022-02-21 12:36:43 +01:00
c43f135571 Update nw to max(nd, 1) (#6714) Glenn Jocher 2022-02-20 21:47:46 +01:00
4de8b24881 Suppress torch AMP-CPU warnings (#6706) Glenn Jocher 2022-02-19 16:48:33 +01:00
de9c25b35e Use export_formats() in export.py (#6705) Glenn Jocher 2022-02-19 16:08:33 +01:00
a297efc383 Edge TPU inference fix (#6686) Raffaele GallieraandGlenn Jocher 2022-02-19 08:10:07 -06:00
0365379016 Fix floating point in number of workers nw (#6701) Samuel Yvon 2022-02-18 15:06:21 -05:00
56697c43ff Fix CoreML P6 inference (#6700) Glenn Jocherandpre-commit-ci[bot] 2022-02-18 20:38:33 +01:00
3670d91b05 Fix --evolve --bucket gs://... (#6698) Glenn Jocher 2022-02-18 19:15:37 +01:00
66e5d794c7 Fix TF exports >= 2GB (#6292)
Jiacong Fang pre-commit-ci[bot]Glenn Jocher
2022-02-18 23:36:45 +08:00
07221f1591 FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6677) Glenn Jocher 2022-02-17 16:05:30 +01:00
18b5081166 FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6673) Glenn Jocher 2022-02-17 15:42:47 +01:00
25456284b4 Update Dockerfile reorder installs (#6672) Glenn Jocher 2022-02-17 15:10:19 +01:00
09d263b376 FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6671) Glenn Jocher 2022-02-17 14:51:53 +01:00
bb054d5f55 FROM nvcr.io/nvidia/pytorch:22.01-py3 (#6670) Glenn Jocher 2022-02-17 14:20:42 +01:00
9a8ebe671c Update Dockerfile torch==1.10.2+cu113 (#6669) Glenn Jocher 2022-02-17 13:24:44 +01:00
2e5c67e537 Robust scipy.cluster.vq.kmeans too few points (#6668) Glenn Jocher 2022-02-17 12:55:03 +01:00
7b80545e8e Add --cache val (#6663) Glenn Jocher 2022-02-17 00:47:42 +01:00
1ff43702a8 Validate with 2x --workers single-GPU/CPU fix (#6659) Glenn Jocher 2022-02-16 14:50:37 +01:00
6769021901 Validate with 2x --workers (#6658) Glenn Jocher 2022-02-16 14:44:44 +01:00
ee6c70ae43 Fix yolov3.yaml remove list (#6655) Glenn Jocher 2022-02-16 10:13:05 +01:00
ca0a00784d Fixed wandb logger KeyError (#6637) imyhxy 2022-02-15 00:08:36 +08:00
538930023c Fix ConfusionMatrix scale vmin=0.0 (#6638) Glenn Jocher 2022-02-14 14:54:55 +01:00
a45e472358 YOLOv5 Export Benchmarks (#6613) Glenn Jocherandpre-commit-ci[bot] 2022-02-12 16:05:43 +01:00
96d8f86085 W&B: don't log media in evolve (#6617) Ayush Chaurasia 2022-02-12 17:34:50 +05:30
51f7750aad W&B: Improve resume stability (#6611) Ayush ChaurasiaandGlenn Jocher 2022-02-12 17:33:09 +05:30
3eef80457f Add YOLOv5n to Reproduce section (#6619) Glenn Jocher 2022-02-11 21:48:01 +01:00
b40bdd6b2d YouTube dependency fix youtube_dl==2020.12.2 (#6612) Glenn Jocher 2022-02-11 13:46:20 +01:00
cb2ad9f685 Fix ROOT / data when running W&B log_dataset() (#6606) or-toledano 2022-02-10 17:17:47 +02:00
c21da596f3 Fix hyp_evolve.yaml indexing bug (#6604) Glenn Jocher 2022-02-10 14:51:29 +01:00
a5c9057dcc Update train.py Glenn Jocher 2022-02-10 14:15:18 +01:00
364b7c5111 Fix YouTube dislike button bug in pafy package (#6603) Glenn Jocher 2022-02-10 13:13:44 +01:00
9cf80b7f60 return opt from train.run() (#6581) Christoph Fürbacher 2022-02-09 15:11:52 +01:00
9c513ca629 Add DATASETS_DIR global in general.py (#6578) Glenn Jocher 2022-02-08 22:20:44 +01:00
f40854b61b Allow custom --evolve project names (#6567) Matthias VadcardandGlenn Jocher 2022-02-08 16:13:45 +01:00
f627bc53f8 Fix plot_labels() colored histogram bug (#6574) Glenn Jocher 2022-02-08 14:03:50 +01:00
22da63e30f Fix zero-export handling with if any(f): (#6569) Glenn Jocher 2022-02-08 12:20:39 +01:00
006eb40ab7 Improved AutoBatch DDP error message (#6568) Glenn Jocher 2022-02-08 11:45:30 +01:00
b57abb17f2 Move trainloader functions to class methods (#6559) Glenn Jocher 2022-02-07 18:59:26 +01:00
dc7e093071 Edge TPU TF imports fix (#6542) Glenn Jocher 2022-02-05 15:45:44 +01:00
cba4303d32 Fix 6 Flake8 issues (#6541) Jirka BorovecandGlenn Jocher 2022-02-05 15:22:59 +01:00
e1a6a0b100 flake8: code meanings (#6481) Jirka Borovec 2022-02-04 19:54:34 +01:00
aff0281969 Load checkpoint on CPU instead of on GPU (#6516) bilzardandGlenn Jocher 2022-02-05 02:58:18 +09:00
8fcdf3b60b Fixing minor multi-streaming issues with TensoRT engine (#6504) greg2451andpre-commit-ci[bot] 2022-02-04 18:19:37 +01:00
079b36d72b Edge TPU tf.lite.experimental.load_delegate fix (#6536) Glenn Jocherandpre-commit-ci[bot] 2022-02-04 09:30:01 +01:00
9bc72a3ac2 Edge TPU export 'list index out of range' fix (#6533) Glenn Jocher 2022-02-04 00:04:06 +01:00
8d05716b66 Edge TPU compiler sudo fix (#6531) Glenn Jocher 2022-02-03 20:40:09 +01:00
a82292ec53 Social icons after text (#6473) Glenn Jocherandpre-commit-ci[bot] 2022-02-03 19:55:19 +01:00
c3e599cfda Update val_batch*.jpg for Chinese fonts (#6526) Glenn Jocherandpre-commit-ci[bot] 2022-02-03 19:09:24 +01:00
cb40c9afda TODO issues exempt from stale action (#6530) Glenn Jocher 2022-02-03 18:11:28 +01:00
19e0208fc9 Update hyp.scratch-high.yaml (#6525) Glenn Jocher 2022-02-03 12:15:13 +01:00
b73c62ebc5 W&B: Remember batchsize on resuming (#6512) Ayush ChaurasiaandGlenn Jocher 2022-02-03 15:59:52 +05:30
4c40933266 Suppress export.run() TracerWarnings (#6499) Glenn Jocher 2022-02-01 23:52:50 +01:00
842d049e1b Suppress torch.jit.TracerWarning on export (#6498) Glenn Jocher 2022-02-01 22:59:26 +01:00
77977e0791 CoreML inference fix list() -> sorted() (#6496) Glenn Jocher 2022-02-01 22:34:15 +01:00
5e4ff195b2 Improved export.py usage examples (#6495) Glenn Jocher 2022-02-01 22:06:29 +01:00
b884ea36c4 Simplify TF normalized to pixels (#6494) Glenn Jocher 2022-02-01 21:17:56 +01:00
6445a8137e Resolve dataset paths (#6489) Glenn Jocher 2022-02-01 15:54:51 +01:00
7539cd75c3 Add Product Hunt social media icon (#6464) Glenn Jocher 2022-01-28 20:23:17 +01:00
d8b5beb0b0 Fix2 select_device() for Multi-GPU (#6461) Glenn Jocher 2022-01-28 08:18:01 +01:00
856d4e5733 Fix select_device() for Multi-GPU (#6434) Glenn Jocher 2022-01-26 23:47:00 +01:00
fe7de6a82d Revert "Remove dataset_stats() autodownload capability (#6303)" (#6442) Glenn Jocher 2022-01-26 19:10:59 +01:00
3b7ac28ed1 Add *.asf video support (#6436) toschi23 2022-01-26 14:26:02 +01:00
d5966c93f1 Namespace VERBOSE env variable to YOLOv5_VERBOSE (#6428) Glenn Jocher 2022-01-25 14:57:27 -08:00
ff8646cdea Update workflows (#6427) Glenn Jocher 2022-01-25 14:33:22 -08:00
16563ac5b5 Prefer tflite_runtime for TFLite inference if installed (#6406) Motoki KimuraandGlenn Jocher 2022-01-26 00:24:24 +09:00
ed9bac8392 Rename logger from 'utils.logger' to 'yolov5' (#6421) Jonathan SamelsonandGlenn Jocher 2022-01-25 16:21:06 +01:00
cfecd903a3 Update greetings.yaml email address (#6412) Glenn Jocher 2022-01-24 15:28:52 -08:00
482af479c0 Add detect.py GIF video inference (#6410) Glenn Jocher 2022-01-24 13:11:11 -08:00

... 10 11 12 13 14 ...