205 Commits (96e36a7c913e2433446ff410a4cf60041010a524)

Author SHA1 Message Date
  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
  Glenn Jocher 0f395b3e3b
YOLOv5 v5.0 Release patch 1 (#2764) 3 years ago
  Glenn Jocher b5de52c4cd
torch.cuda.amp bug fix (#2750) 3 years ago
  Glenn Jocher fca5e2a48f
autocast enable=torch.cuda.is_available() (#2748) 3 years ago
  Glenn Jocher c03d590320
Add Hub results.pandas() method (#2725) 3 years ago
  Glenn Jocher ec8979f1d2
Updated filename attributes for YOLOv5 Hub BytesIO (#2718) 3 years ago
  Glenn Jocher 74276d5189
Updated filename attributes for YOLOv5 Hub results (#2708) 3 years ago
  Glenn Jocher 17300a4c7b
autoShape forward im = np.asarray(im) # to numpy (#2689) 3 years ago
  Glenn Jocher 2af059c0d8
PyTorch Hub model.save() increment as runs/hub/exp (#2684) 3 years ago
  Ding Yiwei 1148e2ea63
Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 3 years ago
  Glenn Jocher 2bf34f50fd
PyTorch Hub amp.autocast() inference (#2641) 3 years ago
  Glenn Jocher 005d7a8c54
Update Detections() self.n comment (#2620) 3 years ago
  Glenn Jocher a57f23d18b
Update detections() self.t = tuple() (#2617) 3 years ago
  Glenn Jocher 1c132a1f94
Update Detections() times=None (#2570) 3 years ago