232 Commits (3beb871ba4558c9e720388e6632798c4eb267d4f)

Author SHA1 Message Date
  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
  Glenn Jocher 525f4f86a9
Add --optimize argument (#3093) 3 years ago
  Glenn Jocher 251aeafcb1
Update P5 + P6 model ensembling (#3082) 3 years ago
  Glenn Jocher e97d129db4
Update export.py with --train mode argument (#3066) 3 years ago
  Glenn Jocher 3ef3a95cfa
Do not optimize CoreML TorchScript model (#3055) 3 years ago
  jylink b292837e36
Fix ONNX export using --grid --simplify --dynamic simultaneously (#2982) 3 years ago
  Hodovo e2a80c6c0f
Add support for FP16 (half) to export.py (#3010) 3 years ago
  jluntamazon 41f5cc5637
YOLOv5 AWS Inferentia Inplace compatibility updates (#2953) 3 years ago
  Glenn Jocher 57812df68c
New Colors() class (#2963) 3 years ago
  Glenn Jocher 4200674a13
Add yolov5/ to sys.path() for *.py subdir exec (#2949) 3 years ago
  Glenn Jocher 4890499344
Improved yolo.py profiling (#2940) 3 years ago
  NanoCode012 c0d3f80544
Add verbose option to pytorch hub models (#2926) 3 years ago
  Glenn Jocher 1b1ab4cca2
Add file_size() function (#2911) 3 years ago
  Glenn Jocher 646386ff09
Update export.py for 2 dry runs (#2910) 3 years ago
  Glenn Jocher a2a514dec8
Update export.py (#2909) 3 years ago
  Glenn Jocher 28db237639
Default optimize_for_mobile() on TorchScript models (#2908) 3 years ago
  Glenn Jocher cdb678f418
Update yolo.py (#2899) 3 years ago
  Glenn Jocher f7bc685c2c
Implement yaml.safe_load() (#2876) 3 years ago
  Burhan c949fc86d1
Detection cropping+saving feature addition for detect.py and PyTorch Hub (#2827) 3 years ago
  Tim Stokman 1df8c6c963
Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional (#2856) 3 years ago
  Glenn Jocher c15e25c40f
PyTorch Hub cv2 .save() .show() bug fix (#2831) 3 years ago
  Glenn Jocher 1f3e482bce
ONNX Simplifier (#2815) 3 years ago