6 Commits (2683b180795c134b7bcdcebd515fac8c0e9cc7a6)

Autor SHA1 Mensaje Fecha
  Glenn Jocher 6e4358f3f3
Add `export.py` ONNX inference suggestion (#4146) hace 3 años
  Glenn Jocher 264be1a616
Rename `opset_version` to `opset` (#4135) hace 3 años
  Glenn Jocher 3bef77f5cb
Addition refactor `export.py` (#4089) hace 3 años
  Glenn Jocher 442a7abdf2
Refactor `export.py` (#4080) hace 3 años
  fcakyon 0e2d0d54d7
Fix typo (#3729) hace 3 años
  Glenn Jocher 1f69d12591
Update 4 main ops for paths and .run() (#3715) hace 3 años
  Glenn Jocher fad27c0046
Update DDP for `torch.distributed.run` with `gloo` backend (#3680) hace 3 años
  lb-desupervised bfb2276b1d
Slightly modify CLI execution (#3687) hace 3 años
  Glenn Jocher 53ed872c28
Update export.py, yolo.py `sys.path.append()` (#3579) hace 3 años
  Glenn Jocher 0e5cfdbea7
Refactor models/export.py arguments (#3564) hace 3 años
  Glenn Jocher f3c3d2ce5d
Merge `develop` branch into `master` (#3518) hace 3 años
  Yonghye Kwon 8d1ddc93c7
Earlier `assert` for cpu and half option (#3508) hace 3 años
  Yonghye Kwon a1c3572bc9
Explicit `model.eval()` call `if opt.train=False` (#3475) hace 3 años
  Sam_S 044daafd9d
Add `output_names` argument for ONNX export with dynamic axes (#3456) hace 3 años
  WangChaofeng ef4d53818d
ONNX export in .train() mode fix (#3362) hace 3 años
  Glenn Jocher 7b36e38cf8
Check CoreML models.train() mode (#3262) hace 3 años
  Cristi Fati 9ab561dbfc
Parameterize ONNX `--opset-version` (#3154) hace 3 años
  Cristi Fati d9b4e6b748
Add `--include torchscript onnx coreml` argument (#3137) hace 3 años
  Glenn Jocher 525f4f86a9
Add --optimize argument (#3093) hace 3 años
  Glenn Jocher e97d129db4
Update export.py with --train mode argument (#3066) hace 3 años
  Glenn Jocher 3ef3a95cfa
Do not optimize CoreML TorchScript model (#3055) hace 3 años
  jylink b292837e36
Fix ONNX export using --grid --simplify --dynamic simultaneously (#2982) hace 3 años
  Hodovo e2a80c6c0f
Add support for FP16 (half) to export.py (#3010) hace 3 años
  Glenn Jocher 4200674a13
Add yolov5/ to sys.path() for *.py subdir exec (#2949) hace 3 años
  Glenn Jocher 1b1ab4cca2
Add file_size() function (#2911) hace 3 años
  Glenn Jocher 646386ff09
Update export.py for 2 dry runs (#2910) hace 3 años
  Glenn Jocher a2a514dec8
Update export.py (#2909) hace 3 años
  Glenn Jocher 28db237639
Default optimize_for_mobile() on TorchScript models (#2908) hace 3 años
  Tim Stokman 1df8c6c963
Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional (#2856) hace 3 años
  Glenn Jocher 1f3e482bce
ONNX Simplifier (#2815) hace 3 años
  Glenn Jocher 0f395b3e3b
YOLOv5 v5.0 Release patch 1 (#2764) hace 3 años
  Jan Hajek 7a0a81fd1d
GPU export options (#2297) hace 3 años
  Aditya Lohia 95aefea493
Dynamic ONNX engine generation (#2208) hace 3 años
  Glenn Jocher 6bd5e8bca7
nn.SiLU() export support (#1713) hace 3 años
  Glenn Jocher d45e349a1e
Rearrange export input after checks (#1118) hace 4 años
  Glenn Jocher 883924d9dc classifier, export, torch seed updates hace 4 años
  Glenn Jocher a12698f4c9
Update ci-testing.yml reflecting sys.path additions (#869) hace 4 años
  PetrDvoracek d55c326dda
export.py folder sys.path addition (#1052) hace 4 años
  Glenn Jocher d68179953d Update export.py hace 4 años
  Glenn Jocher 1e7674d4dd Update export.py --img-size comment hace 4 años
  Glenn Jocher 4fb8cb353f Update export.py with v3.0 Hardswish() support (#831) hace 4 años
  Glenn Jocher 4d7f222f73 Update export.py with v3.0 Hardswish() support hace 4 años
  Glenn Jocher a8751e50de model.fuse() fix for export.py (#827) hace 4 años
  Glenn Jocher b42e8a531b optimize imports hace 4 años
  NanoCode012 0892c44bc4
Fix Logging (#719) hace 4 años
  Jirka Borovec d5b6416c87
Explicit Imports (#498) hace 4 años
  Glenn Jocher 9006b85d21 update export.py torchscript file extension hace 4 años
  Laurentiu Diaconu a9918fbf2b updated coreml conversion to do pixel scaling between 0.0 and 1.0 hace 4 años
  Glenn Jocher 659ad748c8 update get_voc.sh hace 4 años
  Glenn Jocher 7bd7b2c090 update export.py hace 4 años