Jirka Borovec
|
ed887b5976
|
Add pre-commit CI actions (#4982)
* define pre-commit
* add CI code
* configure
* apply pre-commit
* fstring
* apply MD
* pre-commit
* Update torch_utils.py
* Update print strings
* notes
* Cleanup code-format.yml
* Update setup.cfg
* Update .pre-commit-config.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
3 年前 |
Glenn Jocher
|
c3a93d783d
|
Add TensorFlow formats to `export.py` (#4479)
* Initial commit
* Remove unused export_torchscript return
* ROOT variable
* Add prefix to fcn arg
* fix ROOT
* check_yaml into run()
* interim fixes
* imgsz=(320, 320)
* Hardcode tf_raw_resize False
* Finish opt elimination
* Update representative_dataset_gen()
* Update export.py with TF methods
* SiLU and GraphDef fixes
* file_size() directory handling feature
* export fixes
* add lambda: to representative_dataset
* Detect training False default
* Fuse false for TF models
* Embed agnostic NMS arguments
* Remove lambda
* TensorFlow.js export success
* Add pb to Usage
* Add *_tfjs_model/ to ignore files
* prepend YOLOv5 to function headers
* Remove end --- comments
* parameterize tfjs export pb file
* update run() data default /ROOT
* update --include help
* update imports
* return ct_model
* Consolidate TFLite export
* pb prerequisite to tfjs
* TF modules CamelCase
* Remove exports from tf.py and cleanup
* pass agnostic NMS arguments
* CI
* CI
* ignore *_web_model/
* Add tensorflow to CI dependencies
* CI tensorflow-cpu
* Update requirements.txt
* Remove tensorflow check_requirement
* CI coreml tfjs
* export only onnx torchscript
* reorder exports torchscript first
|
3 年前 |
Glenn Jocher
|
6c554b75ea
|
Add TF and TFLite models to `.gitignore` (#4747)
|
3 年前 |
imyhxy
|
f64fab5825
|
Fixed 'meta' and 'hyp' may out of order when using evolve (#4657)
* Fixed 'meta' and 'hyp' may out of order when using evolve
* Update gitignore
|
3 年前 |
Glenn Jocher
|
e78aeac973
|
Evolve in CSV format (#4307)
* Update evolution to CSV format
* Update
* Update
* Update
* Update
* Update
* reset args
* reset args
* reset args
* plot_results() fix
* Cleanup
* Cleanup2
|
3 年前 |
Glenn Jocher
|
96e36a7c91
|
New CSV Logger (#4148)
* New CSV Logger
* cleanup
* move batch plots into Logger
* rename comment
* Remove total loss from progress bar
* mloss :-1 bug fix
* Update plot_results()
* Update plot_results()
* plot_results bug fix
|
3 年前 |
Glenn Jocher
|
fdc22398fa
|
Create `data/hyps` directory (#3747)
|
3 年前 |
SergioSanchezMontesUAM
|
2c99560a98
|
Update .gitignore datasets dir (#1577)
|
3 年前 |
Glenn Jocher
|
9c91aeae10
|
W&B logging add hyperparameters (#1399)
* W&B logging add hyperparameters
* hyp bug fix and image logging updates
* if plots and wandb:
* cleanup
* wandb/ gitignore add
* cleanup 2
* cleanup 3
* move wandb import to top of file
* wandb evolve
* update import
* wandb.run.finish()
* default anchors: 3
|
4 年前 |
Glenn Jocher
|
4821d076e2
|
Increment train, test, detect runs/ (#1322)
* Increment train, test, detect runs/
* Update ci-testing.yml
* inference/images to data/images
* move images
* runs/exp to runs/train/exp
* update 'results saved to %s' str
|
4 年前 |
Glenn Jocher
|
5b8cf247af
|
venv* ignore update
|
4 年前 |
edurenye
|
1c86c2cb66
|
Add torchscript files to gitignore
|
4 年前 |
Glenn Jocher
|
1e84a23f38
|
initial commit
|
4 年前 |