232 Commits (3beb871ba4558c9e720388e6632798c4eb267d4f)

Author SHA1 Message Date
  Glenn Jocher 2da6444c92
Fix for `python models/yolo.py --profile` (#4541) 3 years ago
  Jiacong Fang 808bcad3bb
Add TensorFlow and TFLite export (#1127) 3 years ago
  Glenn Jocher 01cdb7671b
Add `SPPF()` layer (#4420) 3 years ago
  Glenn Jocher 24bea5e4b7
Standardize headers and docstrings (#4417) 3 years ago
  Glenn Jocher 2da4e7acf7
Merge PIL and OpenCV in `plot_one_box(use_pil=False)` (#4416) 3 years ago
  Glenn Jocher 4e8c81a368
Add `yolov5s-ghost.yaml` (#4412) 3 years ago
  Yuantao Yang e7fc27406a
Fix module count in parse_model (#4379) 3 years ago
  Yonghye Kwon e96c74b5a1
Simpler code for DWConvClass (#4310) 3 years ago
  Glenn Jocher 587c4b4b81
Add `DWConvClass()` (#4274) 3 years ago
  Glenn Jocher 9c6732f61c
Update variables (#4273) 3 years ago
  Glenn Jocher 306fc0119a
Update Autoshape forward header (#4271) 3 years ago
  Glenn Jocher 18f6ba77cf
Suppress torch 1.9.0 max_pool2d() warning (#4227) 3 years ago
  Glenn Jocher b60b62e874
PyCharm reformat (#4209) 3 years ago
  Glenn Jocher e016b15555
Update yolov5-bifpn.yaml (#4208) 3 years ago
  Glenn Jocher 2683b18079
Update Hub Path inputs (#4200) 3 years ago
  Glenn Jocher 1f31b7c503
Create yolov5-bifpn.yaml (#4195) 3 years ago
  Glenn Jocher 5d66e48723
Train from `--data path/to/dataset.zip` feature (#4185) 3 years ago
  Glenn Jocher b1be685005
Module `super().__init__()` (#4065) 3 years ago
  Glenn Jocher f7d8562060
`val.py` refactor (#4053) 3 years ago
  Glenn Jocher 9dd33fd20f
AutoShape PosixPath support (#4047) 3 years ago
  Glenn Jocher 720aaa65c8
Rename `test.py` to `val.py` (#4000) 3 years ago
  Glenn Jocher d204a61834
Alert (no detections) (#3984) 3 years ago
  Glenn Jocher b3dabdcc38
Update `probability` to `p` (#3980) 3 years ago
  KEN 647223a7a8
`Ensemble()` visualize fix (#3973) 3 years ago
  jmiranda-laplateforme 588094eb7a
Update `setattr()` default for Hub PIL images (#3923) 3 years ago
  Glenn Jocher 87b094bcbc
Feature visualization update (#3920) 3 years ago
  Glenn Jocher 3c3f8fbd5d
Improved BGR2RGB speeds (#3880) 3 years ago
  Glenn Jocher 81b31824f5
Models `*.yaml` reformat (#3875) 3 years ago
  Valentin Aliferov 831773f5a2
Add EXIF rotation to YOLOv5 Hub inference (#3852) 3 years ago
  Glenn Jocher 25d1f2932c
Plot `AutoShape()` detections in ascending order (#3843) 3 years ago
  Glenn Jocher 57c5d02bbe
Concise `TransformerBlock()` (#3821) 3 years ago
  Zigarss 20d45aa4f1
Add feature map visualization (#3804) 3 years ago
  Glenn Jocher 1f69d12591
Update 4 main ops for paths and .run() (#3715) 3 years ago
  Glenn Jocher fad27c0046
Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 3 years ago
  lb-desupervised bfb2276b1d
Slightly modify CLI execution (#3687) 3 years ago
  Glenn Jocher 53ed872c28
Update export.py, yolo.py `sys.path.append()` (#3579) 3 years ago
  Glenn Jocher 0e5cfdbea7
Refactor models/export.py arguments (#3564) 3 years ago
  Glenn Jocher f3c3d2ce5d
Merge `develop` branch into `master` (#3518) 3 years ago
  Yonghye Kwon 8d1ddc93c7
Earlier `assert` for cpu and half option (#3508) 3 years ago
  Yonghye Kwon a1c3572bc9
Explicit `model.eval()` call `if opt.train=False` (#3475) 3 years ago
  Sam_S 044daafd9d
Add `output_names` argument for ONNX export with dynamic axes (#3456) 3 years ago
  chocosaj 3cb9ad4fc4
Update FLOPs description (#3422) 3 years ago
  Glenn Jocher ba6f3f974b
Enable direct `--weights URL` definition (#3373) 3 years ago
  WangChaofeng ef4d53818d
ONNX export in .train() mode fix (#3362) 3 years ago
  Glenn Jocher 1f8d716ec9
yolo.py header (#3347) 3 years ago
  Glenn Jocher 7b36e38cf8
Check CoreML models.train() mode (#3262) 3 years ago
  Adrian Holovaty 3f74cd9ed1
Parameterize max_det + inference default at 1000 (#3215) 3 years ago
  Cristi Fati 9ab561dbfc
Parameterize ONNX `--opset-version` (#3154) 3 years ago
  Yonghye Kwon be86c21c73
rename class autoShape -> AutoShape (#3173) 3 years ago
  Cristi Fati d9b4e6b748
Add `--include torchscript onnx coreml` argument (#3137) 3 years ago