Glenn Jocher
b3dabdcc38
Update `probability` to `p` ( #3980 )
2021-07-12 15:54:43 +02:00
KEN
647223a7a8
`Ensemble()` visualize fix ( #3973 )
...
* fix visualize error
* Revert "fix visualize error"
* add visualise profile
2021-07-11 19:47:08 +02:00
jmiranda-laplateforme
588094eb7a
Update `setattr()` default for Hub PIL images ( #3923 )
...
Fix inference from PIL source.
2021-07-07 16:13:12 +02:00
Glenn Jocher
87b094bcbc
Feature visualization update ( #3920 )
...
* Feature visualization update
* Save to jpg (faster)
* Save to png
2021-07-07 15:41:58 +02:00
Glenn Jocher
3c3f8fbd5d
Improved BGR2RGB speeds ( #3880 )
...
* Update BGR2RGB ops
* speed improvements
* cleanup
2021-07-04 20:12:32 +02:00
Glenn Jocher
81b31824f5
Models `*.yaml` reformat ( #3875 )
2021-07-04 16:55:08 +02:00
Valentin Aliferov
831773f5a2
Add EXIF rotation to YOLOv5 Hub inference ( #3852 )
...
* rotating an image according to its exif tag
* Update common.py
* Update datasets.py
* Update datasets.py
faster
* delete extraneous gpg file
* Update common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-02 13:25:54 +02:00
Glenn Jocher
25d1f2932c
Plot `AutoShape()` detections in ascending order ( #3843 )
2021-06-30 15:10:40 +02:00
Glenn Jocher
57c5d02bbe
Concise `TransformerBlock()` ( #3821 )
2021-06-29 16:03:10 +02:00
Zigarss
20d45aa4f1
Add feature map visualization ( #3804 )
...
* Add feature map visualization
Add a feature_visualization function to visualize the mid feature map of the model.
* Update yolo.py
* remove boolean from forward and reorder if statement
* remove print from forward
* General cleanup
* Indent
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-28 13:18:45 +02:00
Glenn Jocher
1f69d12591
Update 4 main ops for paths and .run() ( #3715 )
...
* Add yolov5/ to path
* rename functions to run()
* cleanup
* rename fix
* CI fix
* cleanup find models/export.py
2021-06-21 17:25:04 +02:00
Glenn Jocher
fad27c0046
Update DDP for `torch.distributed.run` with `gloo` backend ( #3680 )
...
* Update DDP for `torch.distributed.run`
* Add LOCAL_RANK
* remove opt.local_rank
* backend="gloo|nccl"
* print
* print
* debug
* debug
* os.getenv
* gloo
* gloo
* gloo
* cleanup
* fix getenv
* cleanup
* cleanup destroy
* try nccl
* return opt
* add --local_rank
* add timeout
* add init_method
* gloo
* move destroy
* move destroy
* move print(opt) under if RANK
* destroy only RANK 0
* move destroy inside train()
* restore destroy outside train()
* update print(opt)
* cleanup
* nccl
* gloo with 60 second timeout
* update namespace printing
2021-06-19 16:30:25 +02:00
lb-desupervised
bfb2276b1d
Slightly modify CLI execution ( #3687 )
...
* Slightly modify CLI execution
This simple change makes it easier to run the primary functions of this
repo (train/detect/test) from within Python. An object which represents
`opt` can be constructed and fed to the `main` function of each of these
modules, rather than having to call the lower level functions directly,
or run the module as a script.
* Update export.py
Add CLI parsing update for more convenient module usage within Python.
Co-authored-by: Lewis Belcher <lb@desupervised.io>
2021-06-19 12:06:59 +02:00
Glenn Jocher
53ed872c28
Update export.py, yolo.py `sys.path.append()` ( #3579 )
2021-06-10 15:35:22 +02:00
Glenn Jocher
0e5cfdbea7
Refactor models/export.py arguments ( #3564 )
...
* Refactor models/export.py arguments
* cleanup
* cleanup
2021-06-09 22:43:46 +02:00
Glenn Jocher
f3c3d2ce5d
Merge `develop` branch into `master` ( #3518 )
...
* update ci-testing.yml (#3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* update ci-testing.yml (#3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* Enable direct `--weights URL` definition (#3373 )
* Enable direct `--weights URL` definition
@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```
* cleanup
* bug fixes
* weights = attempt_download(weights)
* Update experimental.py
* Update hubconf.py
* return bug fix
* comment mirror
* min_bytes
* Update tutorial.ipynb (#3368 )
add Open in Kaggle badge
* `cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379 )
* Update datasets.py
* comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* COCO evolution fix (#3388 )
* COCO evolution fix
* cleanup
* update print
* print fix
* Create `is_pip()` function (#3391 )
Returns `True` if file is part of pip package. Useful for contextual behavior modification.
```python
def is_pip():
# Is file in a pip package?
return 'site-packages' in Path(__file__).absolute().parts
```
* Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379 )" (#3395 )
This reverts commit 21a9607e00
.
* Update FLOPs description (#3422 )
* Update README.md
* Changing FLOPS to FLOPs.
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
* Parse URL authentication (#3424 )
* Parse URL authentication
* urllib.parse.unquote()
* improved error handling
* improved error handling
* remove %3F
* update check_file()
* Add FLOPs title to table (#3453 )
* Suppress jit trace warning + graph once (#3454 )
* Suppress jit trace warning + graph once
Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.
* Update train.py
* Update MixUp augmentation `alpha=beta=32.0` (#3455 )
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
* Add `timeout()` class (#3460 )
* Add `timeout()` class
* rearrange order
* Faster HSV augmentation (#3462 )
remove datatype conversion process that can be skipped
* Add `check_git_status()` 5 second timeout (#3464 )
* Add check_git_status() 5 second timeout
This should prevent the SSH Git bug that we were discussing @KalenMike
* cleanup
* replace timeout with check_output built-in timeout
* Improved `check_requirements()` offline-handling (#3466 )
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
* Add `output_names` argument for ONNX export with dynamic axes (#3456 )
* Add output names & dynamic axes for onnx export
Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3
* use first output only + cleanup
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Revert FP16 `test.py` and `detect.py` inference to FP32 default (#3423 )
* fixed inference bug ,while use half precision
* replace --use-half with --half
* replace space and PEP8 in detect.py
* PEP8 detect.py
* update --half help comment
* Update test.py
* revert space
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Add additional links/resources to stale.yml message (#3467 )
* Update stale.yml
* cleanup
* Update stale.yml
* reformat
* Update stale.yml HUB URL (#3468 )
* Stale `github.actor` bug fix (#3483 )
* Explicit `model.eval()` call `if opt.train=False` (#3475 )
* call model.eval() when opt.train is False
call model.eval() when opt.train is False
* single-line if statement
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* check_requirements() exclude `opencv-python` (#3495 )
Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494 .
* Earlier `assert` for cpu and half option (#3508 )
* early assert for cpu and half option
early assert for cpu and half option
* Modified comment
Modified comment
* Update tutorial.ipynb (#3510 )
* Reduce test.py results spacing (#3511 )
* Update README.md (#3512 )
* Update README.md
Minor modifications
* 850 width
* Update greetings.yml
revert greeting change as PRs will now merge to master.
Co-authored-by: Piotr Skalski <SkalskiP@users.noreply.github.com>
Co-authored-by: SkalskiP <piotr.skalski92@gmail.com>
Co-authored-by: Peretz Cohen <pizzaz93@users.noreply.github.com>
Co-authored-by: tudoulei <34886368+tudoulei@users.noreply.github.com>
Co-authored-by: chocosaj <chocosaj@users.noreply.github.com>
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Sam_S <SamSamhuns@users.noreply.github.com>
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: edificewang <609552430@qq.com>
2021-06-08 10:22:10 +02:00
WangChaofeng
ef4d53818d
ONNX export in .train() mode fix ( #3362 )
2021-05-27 14:10:14 +02:00
Glenn Jocher
1f8d716ec9
yolo.py header ( #3347 )
2021-05-26 12:06:08 +02:00
Glenn Jocher
7b36e38cf8
Check CoreML models.train() mode ( #3262 )
...
* Check CoreML models.train() mode
* Update export.py
2021-05-20 15:22:36 +02:00
Adrian Holovaty
3f74cd9ed1
Parameterize max_det + inference default at 1000 ( #3215 )
...
* Added max_det parameters in various places
* 120 character line
* PEP8
* 120 character line
* Update inference default to 1000 instances
* Update inference default to 1000 instances
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-17 22:47:44 +02:00
Cristi Fati
9ab561dbfc
Parameterize ONNX `--opset-version` ( #3154 )
2021-05-16 16:13:03 +02:00
Yonghye Kwon
be86c21c73
rename class autoShape -> AutoShape ( #3173 )
...
* rename class autoShape -> AutoShape
follow other class naming convention
* rename class autoShape -> AutoShape
follow other classes' naming convention
* rename class autoShape -> AutoShape
2021-05-16 15:46:45 +02:00
Cristi Fati
d9b4e6b748
Add `--include torchscript onnx coreml` argument ( #3137 )
...
* Allow users to skip exporting in formats that they don't care about
* Correct comments
* Update export.py
renamed --skip-format to --exclude
* Switched format from exclude to include (as instructed by @glenn-jocher)
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-12 19:46:32 +02:00
Glenn Jocher
525f4f86a9
Add --optimize argument ( #3093 )
...
Fix for c++ runtime errors in https://github.com/ultralytics/yolov5/issues/2973
2021-05-10 12:13:44 +02:00
Glenn Jocher
251aeafcb1
Update P5 + P6 model ensembling ( #3082 )
2021-05-08 19:06:12 +02:00
Glenn Jocher
e97d129db4
Update export.py with --train mode argument ( #3066 )
2021-05-07 13:10:07 +02:00
Glenn Jocher
3ef3a95cfa
Do not optimize CoreML TorchScript model ( #3055 )
2021-05-06 17:23:33 +02:00
jylink
b292837e36
Fix ONNX export using --grid --simplify --dynamic simultaneously ( #2982 )
...
* Update yolo.py
* Update export.py
* fix export grid
* Update export.py, remove detect export attribute
* rearrange if order
* remove --grid, default inplace=False
* rename exp_dynamic to onnx_dynamic, comment
* replace bs with 1 in anchor_grid[i] index 0
* Update export.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-03 19:01:29 +02:00
Hodovo
e2a80c6c0f
Add support for FP16 (half) to export.py ( #3010 )
...
* Added support for fp16 (half) to export.py
* minimize code additions
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-02 22:42:33 +02:00
jluntamazon
41f5cc5637
YOLOv5 AWS Inferentia Inplace compatibility updates ( #2953 )
...
* Added flag to enable/disable all inplace and assignment operations
* Removed shape print statements
* Scope Detect/Model import to avoid circular dependency
* PEP8
* create _descale_pred()
* replace lost space
* replace list with tuple
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-30 12:54:48 +02:00
Glenn Jocher
57812df68c
New Colors() class ( #2963 )
2021-04-28 16:05:14 +02:00
Glenn Jocher
4200674a13
Add yolov5/ to sys.path() for *.py subdir exec ( #2949 )
...
* Add yolov5/ to sys.path() for *.py subdir exec
* Update export.py
2021-04-27 17:02:07 +02:00
Glenn Jocher
4890499344
Improved yolo.py profiling ( #2940 )
...
* Improved yolo.py profiling
Improved column order and labelling.
* Update yolo.py
2021-04-26 23:58:21 +02:00
NanoCode012
c0d3f80544
Add verbose option to pytorch hub models ( #2926 )
...
* Add verbose and update print to logging
* Fix positonal param
* Revert auto formatting changes
* Update hubconf.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-25 22:01:05 +02:00
Glenn Jocher
1b1ab4cca2
Add file_size() function ( #2911 )
...
* Add file_size() function
* Update export.py
2021-04-24 01:31:11 +02:00
Glenn Jocher
646386ff09
Update export.py for 2 dry runs ( #2910 )
...
* Update export.py for 2 dry runs
* Update export.py
2021-04-24 00:10:38 +02:00
Glenn Jocher
a2a514dec8
Update export.py ( #2909 )
2021-04-23 23:50:02 +02:00
Glenn Jocher
28db237639
Default optimize_for_mobile() on TorchScript models ( #2908 )
...
Per https://pytorch.org/tutorials/recipes/script_optimized.html this should improve performance on torchscript models (and maybe coreml models also since coremltools operates on a torchscript model input, though this still requires testing).
2021-04-23 21:21:58 +02:00
Glenn Jocher
cdb678f418
Update yolo.py ( #2899 )
2021-04-22 19:27:21 +02:00
Glenn Jocher
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
...
* Implement yaml.safe_load()
* yaml.safe_dump()
2021-04-21 14:34:45 +02:00
Burhan
c949fc86d1
Detection cropping+saving feature addition for detect.py and PyTorch Hub ( #2827 )
...
* Update detect.py
* Update detect.py
* Update greetings.yml
* Update cropping
* cleanup
* Update increment_path()
* Update common.py
* Update detect.py
* Update detect.py
* Update detect.py
* Update common.py
* cleanup
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-20 23:51:08 +02:00
Tim Stokman
1df8c6c963
Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional ( #2856 )
...
* Ensure dynamic export works succesfully, onnx simplifier optional
* Update export.py
* add dashes
Co-authored-by: Tim <tim.stokman@hal24k.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-20 13:54:03 +02:00
Glenn Jocher
c15e25c40f
PyTorch Hub cv2 .save() .show() bug fix ( #2831 )
...
* PyTorch Hub cv2 .save() .show() bug fix
cv2.rectangle() was failing on non-contiguous np array inputs. This checks for contiguous arrays and applies is necessary:
```python
imgs[i] = im if im.data.contiguous else np.ascontiguousarray(im) # update
```
* Update plots.py
```python
assert im.data.contiguous, 'Image not contiguous. Apply np.ascontiguousarray(im) to plot_on_box() input image.'
```
* Update hubconf.py
Expand CI tests to OpenCV image.
2021-04-18 13:47:40 +02:00
Glenn Jocher
1f3e482bce
ONNX Simplifier ( #2815 )
...
* ONNX Simplifier
Add ONNX Simplifier to ONNX export pipeline in export.py. Will auto-install onnx-simplifier if onnx is installed but onnx-simplifier is not.
* Update general.py
2021-04-16 14:03:27 +02:00
Glenn Jocher
0f395b3e3b
YOLOv5 v5.0 Release patch 1 ( #2764 )
...
* torch.jit.trace(model, img, strict=False)
* Update check_file()
* Update hubconf.py
* Update README.md
2021-04-11 23:11:43 +02:00
Glenn Jocher
b5de52c4cd
torch.cuda.amp bug fix ( #2750 )
...
PR https://github.com/ultralytics/yolov5/pull/2725 introduced a very specific bug that only affects multi-GPU trainings. Apparently the cause was using the torch.cuda.amp decorator in the autoShape forward method. I've implemented amp more traditionally in this PR, and the bug is resolved.
2021-04-09 18:19:49 +02:00
Glenn Jocher
fca5e2a48f
autocast enable=torch.cuda.is_available() ( #2748 )
2021-04-09 13:34:49 +02:00
Glenn Jocher
c03d590320
Add Hub results.pandas() method ( #2725 )
...
* Add Hub results.pandas() method
New method converts results from torch tensors to pandas DataFrames with column names.
This PR may partially resolve issue https://github.com/ultralytics/yolov5/issues/2703
```python
results = model(imgs)
print(results.pandas().xyxy[0])
xmin ymin xmax ymax confidence class name
0 57.068970 391.770599 241.383545 905.797852 0.868964 0 person
1 667.661255 399.303589 810.000000 881.396667 0.851888 0 person
2 222.878387 414.774231 343.804474 857.825073 0.838376 0 person
3 4.205386 234.447678 803.739136 750.023376 0.658006 5 bus
4 0.000000 550.596008 76.681190 878.669922 0.450596 0 person
```
* Update comments
torch example input now shown resized to size=640 and also now a multiple of P6 stride 64 (see https://github.com/ultralytics/yolov5/issues/2722#issuecomment-814785930 )
* apply decorators
* PEP8
* Update common.py
* pd.options.display.max_columns = 10
* Update common.py
2021-04-07 16:28:07 +02:00
Glenn Jocher
ec8979f1d2
Updated filename attributes for YOLOv5 Hub BytesIO ( #2718 )
...
Fix 2 for 'Model predict with forward will fail if PIL image does not have filename attribute' #2702
2021-04-06 13:18:56 +02:00
Glenn Jocher
74276d5189
Updated filename attributes for YOLOv5 Hub results ( #2708 )
...
Proposed fix for 'Model predict with forward will fail if PIL image does not have filename attribute' #2702
2021-04-05 22:20:09 +02:00
Glenn Jocher
17300a4c7b
autoShape forward im = np.asarray(im) # to numpy ( #2689 )
...
Slight speedup.
2021-04-02 12:36:38 +02:00
Glenn Jocher
2af059c0d8
PyTorch Hub model.save() increment as runs/hub/exp ( #2684 )
...
* PyTorch Hub model.save() increment as runs/hub/exp
This chane will align PyTorch Hub results saving with the existing unified results saving directory structure of
runs/
/train
/detect
/test
/hub
/exp
/exp2
...
* cleanup
2021-04-02 11:55:10 +02:00
Ding Yiwei
1148e2ea63
Add TransformerLayer, TransformerBlock, C3TR modules ( #2333 )
...
* yolotr
* transformer block
* Remove bias in Transformer
* Remove C3T
* Remove a deprecated class
* put the 2nd LayerNorm into the 2nd residual block
* move example model to models/hub, rename to -transformer
* Add module comments and TODOs
* Remove LN in Transformer
* Add comments for Transformer
* Solve the problem of MA with DDP
* cleanup
* cleanup find_unused_parameters
* PEP8 reformat
Co-authored-by: DingYiwei <846414640@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-01 17:26:53 +02:00
Glenn Jocher
2bf34f50fd
PyTorch Hub amp.autocast() inference ( #2641 )
...
I think this should help speed up CUDA inference, as currently models may be running in FP32 inference mode on CUDA devices unnecesarily.
2021-03-28 20:23:40 +02:00
Glenn Jocher
005d7a8c54
Update Detections() self.n comment ( #2620 )
...
```python
self.n = len(self.pred) # number of images (batch size)
```
2021-03-26 21:19:15 +01:00
Glenn Jocher
a57f23d18b
Update detections() self.t = tuple() ( #2617 )
...
* Update detections() self.t = tuple()
Fix multiple results.print() bug.
* Update experimental.py
* Update yolo.py
2021-03-26 13:23:11 +01:00
Glenn Jocher
1c132a1f94
Update Detections() times=None ( #2570 )
...
Fix for results.tolist() method breaking after YOLOv5 Hub profiling PRshttps://github.com/ultralytics/yolov5/pull/2460 https://github.com/ultralytics/yolov5/pull/2459 and
2021-03-23 14:10:47 +01:00
Ayush Chaurasia
e8fc97aa38
Improved W&B integration ( #2125 )
...
* Init Commit
* new wandb integration
* Update
* Use data_dict in test
* Updates
* Update: scope of log_img
* Update: scope of log_img
* Update
* Update: Fix logging conditions
* Add tqdm bar, support for .txt dataset format
* Improve Result table Logger
* Init Commit
* new wandb integration
* Update
* Use data_dict in test
* Updates
* Update: scope of log_img
* Update: scope of log_img
* Update
* Update: Fix logging conditions
* Add tqdm bar, support for .txt dataset format
* Improve Result table Logger
* Add dataset creation in training script
* Change scope: self.wandb_run
* Add wandb-artifact:// natively
you can now use --resume with wandb run links
* Add suuport for logging dataset while training
* Cleanup
* Fix: Merge conflict
* Fix: CI tests
* Automatically use wandb config
* Fix: Resume
* Fix: CI
* Enhance: Using val_table
* More resume enhancement
* FIX : CI
* Add alias
* Get useful opt config data
* train.py cleanup
* Cleanup train.py
* more cleanup
* Cleanup| CI fix
* Reformat using PEP8
* FIX:CI
* rebase
* remove uneccesary changes
* remove uneccesary changes
* remove uneccesary changes
* remove unecessary chage from test.py
* FIX: resume from local checkpoint
* FIX:resume
* FIX:resume
* Reformat
* Performance improvement
* Fix local resume
* Fix local resume
* FIX:CI
* Fix: CI
* Imporve image logging
* (:(:Redo CI tests:):)
* Remember epochs when resuming
* Remember epochs when resuming
* Update DDP location
Potential fix for #2405
* PEP8 reformat
* 0.25 confidence threshold
* reset train.py plots syntax to previous
* reset epochs completed syntax to previous
* reset space to previous
* remove brackets
* reset comment to previous
* Update: is_coco check, remove unused code
* Remove redundant print statement
* Remove wandb imports
* remove dsviz logger from test.py
* Remove redundant change from test.py
* remove redundant changes from train.py
* reformat and improvements
* Fix typo
* Add tqdm tqdm progress when scanning files, naming improvements
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-03-23 00:44:50 +01:00
Glenn Jocher
f813f6dcc8
autoShape() speed profiling update ( #2460 )
2021-03-13 20:00:03 -08:00
Glenn Jocher
569757ecc0
Add autoShape() speed profiling ( #2459 )
...
* Add autoShape() speed profiling
* Update common.py
* Create README.md
* Update hubconf.py
* cleanuip
2021-03-13 19:50:34 -08:00
Jan Hajek
7a0a81fd1d
GPU export options ( #2297 )
...
* option for skip last layer and cuda export support
* added parameter device
* fix import
* cleanup 1
* cleanup 2
* opt-in grid
--grid will export with grid computation, default export will skip grid (same as current)
* default --device cpu
GPU export causes ONNX and CoreML errors.
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-03-06 12:02:10 -08:00
Glenn Jocher
a3ecf0fd64
Anchor override ( #2350 )
2021-03-02 23:08:21 -08:00
Glenn Jocher
fab5085674
EMA bug fix 2 ( #2330 )
...
* EMA bug fix 2
* update
2021-03-01 17:13:34 -08:00
Glenn Jocher
fd96810518
remove TTA 1 pixel offset ( #2325 )
2021-02-28 21:14:08 -08:00
oleg
cbd55da5d2
Update yolo.py ( #2120 )
...
* Avoid mutable state in Detect
* LoadImages() pathlib update (#2140 )
* Unique *.cache filenames fix (#2134 )
* fix #2121
* Update test.py
* Update train.py
* Update autoanchor.py
* Update datasets.py
* Update log_dataset.py
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update train.py test batch_size (#2148 )
* Update train.py
* Update loss.py
* Update train.py (#2149 )
* Linear LR scheduler option (#2150 )
* Linear LR scheduler option
* Update train.py
* Update data-autodownload background tasks (#2154 )
* Update get_coco.sh
* Update get_voc.sh
* Update detect.py (#2167 )
Without this cv2.imshow opens a window but nothing is visible
* Update requirements.txt (#2173 )
* Update utils/datasets.py to support .webp files (#2174 )
Simply added 'webp' as an image format to the img_formats array so that webp image files can be used as training data.
* Changed socket port and added timeout (#2176 )
* PyTorch Hub results.save('path/to/dir') (#2179 )
* YOLOv5 Segmentation Dataloader Updates (#2188 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* GitHub API rate limit fix
* update
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* astuple
* epochs
* update
* update
* ComputeLoss()
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* merge
* merge
* merge
* merge
* update
* update
* update
* update
* commit=tag == tags[-1]
* Update cudnn.benchmark
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* mosaic9
* update
* update
* update
* update
* update
* update
* institute cache versioning
* only display on existing cache
* reverse cache exists booleans
* Created using Colaboratory
* YOLOv5 PyTorch Hub results.save() method retains filenames (#2194 )
* save results with name
* debug
* save original imgs names
* Update common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* TTA augument boxes one pixel shifted in de-flip ud and lr (#2219 )
* TTA augument boxes one pixel shifted in de-flip ud and lr
* PEP8 reformat
Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* LoadStreams() frame loss bug fix (#2222 )
* Update yolo.py channel array (#2223 )
* Add check_imshow() (#2231 )
* Add check_imshow()
* Update general.py
* Update general.py
* Update CI badge (#2230 )
* Add isdocker() (#2232 )
* Add isdocker()
* Update general.py
* Update general.py
* YOLOv5 Hub URL inference bug fix (#2250 )
* Update common.py
* Update common.py
* Update common.py
* Improved hubconf.py CI tests (#2251 )
* Unified hub and detect.py box and labels plotting (#2243 )
* reset head
* Update inference default to multi_label=False (#2252 )
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
* Robust objectness loss balancing (#2256 )
* Created using Colaboratory
* Update minimum stride to 32 (#2266 )
* Dynamic ONNX engine generation (#2208 )
* add: dynamic onnx export
* delete: test onnx inference
* fix dynamic output axis
* Code reduction
* fix: dynamic output axes, dynamic input naming
* Remove fixed axes
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update greetings.yml for auto-rebase on PR (#2272 )
* Update Dockerfile with apt install zip (#2274 )
* FLOPS min stride 32 (#2276 )
Signed-off-by: xiaowo1996 <429740343@qq.com>
* Update README.md
* Amazon AWS EC2 startup and re-startup scripts (#2185 )
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
* Amazon AWS EC2 startup and re-startup scripts (#2282 )
* Update train.py (#2290 )
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
* Improved model+EMA checkpointing (#2292 )
* Enhanced model+EMA checkpointing
* update
* bug fix
* bug fix 2
* always save optimizer
* ema half
* remove model.float()
* model half
* carry ema/model in fp32
* rm model.float()
* both to float always
* cleanup
* cleanup
* Improved model+EMA checkpointing 2 (#2295 )
* Fix labels being missed when image extension appears twice in filename (#2300 )
* W&B entity support (#2298 )
* W&B entity support
* shorten wandb_entity to entity
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Avoid mutable state in Detect
* Update yolo and remove .to(device)
Co-authored-by: Oleg Boiko <oboiko@chegg.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: train255 <thanhdd.it@gmail.com>
Co-authored-by: ab-101 <56578530+ab-101@users.noreply.github.com>
Co-authored-by: Transigent <wbdsmith@optusnet.com.au>
Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
Co-authored-by: Daniel Khromov <danielkhromov@gmail.com>
Co-authored-by: VdLMV <jaap@vdlmv.nl>
Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Yann Defretin <kinoute@gmail.com>
Co-authored-by: Aditya Lohia <64709773+aditya-dl@users.noreply.github.com>
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: xiaowo1996 <429740343@qq.com>
Co-authored-by: Iden Craven <iden.craven@gmail.com>
Co-authored-by: Jan Hajek <toretak@users.noreply.github.com>
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
2021-02-26 15:07:40 -08:00
Aditya Lohia
95aefea493
Dynamic ONNX engine generation ( #2208 )
...
* add: dynamic onnx export
* delete: test onnx inference
* fix dynamic output axis
* Code reduction
* fix: dynamic output axes, dynamic input naming
* Remove fixed axes
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-21 21:50:44 -08:00
Glenn Jocher
c09964c27c
Update inference default to multi_label=False ( #2252 )
...
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
2021-02-19 15:39:09 -08:00
Yann Defretin
5f42643a53
Unified hub and detect.py box and labels plotting ( #2243 )
2021-02-19 12:38:05 -08:00
Glenn Jocher
db28ce61ac
Improved hubconf.py CI tests ( #2251 )
2021-02-19 12:35:38 -08:00
Glenn Jocher
9d87307784
YOLOv5 Hub URL inference bug fix ( #2250 )
...
* Update common.py
* Update common.py
* Update common.py
2021-02-19 11:22:42 -08:00
Glenn Jocher
f8464b4f66
Update yolo.py channel array ( #2223 )
2021-02-15 21:21:53 -08:00
VdLMV
3ff783c18f
TTA augument boxes one pixel shifted in de-flip ud and lr ( #2219 )
...
* TTA augument boxes one pixel shifted in de-flip ud and lr
* PEP8 reformat
Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-15 10:49:22 -08:00
Daniel Khromov
3e560e2fae
YOLOv5 PyTorch Hub results.save() method retains filenames ( #2194 )
...
* save results with name
* debug
* save original imgs names
* Update common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-12 15:37:51 -08:00
Glenn Jocher
404749a33c
PyTorch Hub results.save('path/to/dir') ( #2179 )
2021-02-10 16:10:43 -08:00
Glenn Jocher
9646ca438a
Update plot_study() ( #2112 )
2021-02-01 13:51:11 -08:00
Glenn Jocher
6e6f77be47
Add YOLOv5-P6 models ( #2083 )
2021-01-29 11:57:29 -08:00
Glenn Jocher
6bfa9c2422
GhostConv update ( #2082 )
2021-01-29 11:25:01 -08:00
Glenn Jocher
08d3119e09
Add histogram equalization fcn ( #2049 )
2021-01-26 17:30:42 -08:00
Glenn Jocher
2fc4760257
Update autoshape .print() and .save() ( #2022 )
2021-01-22 16:38:06 -08:00
Glenn Jocher
aac33f87ad
Update inference multiple-counting ( #2019 )
...
* Update inference multiple-counting
* update github check
2021-01-22 15:39:08 -08:00
Abhiram V
4501169547
Update yolo.py with yaml.SafeLoader ( #1970 )
2021-01-18 10:47:00 -08:00
lorne
03ebe6ea14
Add nn.SiLU inplace in attempt_load() ( #1940 )
...
* sleep 3 sec to load data .
* Update train.py
* Add nn.SiLU inplace in attempt_load()
Co-authored-by: wangliang <wangliang@codingapi.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-01-15 09:58:41 -08:00
Glenn Jocher
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
2021-01-10 20:06:11 -08:00
Glenn Jocher
69be8e738f
YOLOv5 v4.0 Release ( #1837 )
...
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update study.png
* update study.png
* Update datasets.py
2021-01-04 19:54:09 -08:00
Glenn Jocher
0b6266f5e0
Update yolov3-tiny.yaml ( #1823 )
2021-01-02 13:01:05 -08:00
NanoCode012
14b0abe2d0
autoShape() default for PyTorch Hub models ( #1692 )
...
* Add autoshape parameter
* Remove autoshape call in ReadMe
* Update hubconf.py
* file/URI inputs and autoshape check passthrough
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-12-26 19:58:26 -08:00
Glenn Jocher
9922c1183f
leaf Variable inplace bug fix ( #1759 )
2020-12-22 17:27:40 -08:00
Glenn Jocher
394d1c89f3
Input channel yaml['ch'] addition ( #1741 )
2020-12-19 10:54:01 -08:00
Glenn Jocher
6bd5e8bca7
nn.SiLU() export support ( #1713 )
2020-12-16 17:55:57 -08:00
Glenn Jocher
799724108f
Update C3 module ( #1705 )
2020-12-15 22:13:08 -08:00
Glenn Jocher
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
...
* Simplify autoshape() post-process
* cleanup
* cleanup
2020-12-09 07:44:06 -08:00
Glenn Jocher
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
...
* Pycocotools best.pt after COCO train
* cleanup
2020-12-06 14:58:33 +01:00
Glenn Jocher
d929bb656c
Implement default class names ( #1609 )
2020-12-06 12:41:37 +01:00
Glenn Jocher
ba48f867ea
Add bias to Classify() ( #1601 )
2020-12-04 15:06:33 +01:00
Glenn Jocher
199c9c7874
Detection() device bug fix ( #1455 )
2020-11-19 13:45:36 +01:00
Glenn Jocher
225845e781
Update test.py for IoU in native image-space ( #1439 )
...
* Update test.py for IoU in native image-space
* remove redundant
* gn to device
* remove output scale_coords
* --img-size correction
* update
* native-space labels
* pred to predn
* remove clip_coords()
2020-11-18 22:50:21 +01:00
Glenn Jocher
97c6a59dd1
PANet anchor order correction ( #1420 )
2020-11-17 10:13:00 +01:00
Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
...
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
2020-11-16 23:09:55 +01:00
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
...
* Improved FLOPS computation
* update comment
2020-11-14 14:39:46 +01:00
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
...
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
2020-11-14 11:50:32 +01:00
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
...
* Create flatten_recursive() helper function
* cleanup
* print torch version
2020-11-09 12:24:11 +01:00
Glenn Jocher
15a10609fe
autoshape() update for PIL greyscale inputs ( #1279 )
...
* autoshape update for PIL greyscale inputs
* autoshape update for PIL greyscale inputs
2020-11-04 14:20:11 +01:00
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
...
* torch 1.7.0 compatibility updates
* add inference verification
2020-10-28 15:03:50 +01:00
Yi_Zhang
f532bc68a9
GhostConv() bug fix ( #1176 )
2020-10-20 13:42:11 +02:00
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
2020-10-15 20:10:08 +02:00
Glenn Jocher
d45e349a1e
Rearrange export input after checks ( #1118 )
...
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
2020-10-11 16:23:36 +02:00
Glenn Jocher
c2403d785c
fuse() bug fix
2020-10-11 00:42:54 +02:00
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
2020-10-06 14:54:02 +02:00
Glenn Jocher
a12698f4c9
Update ci-testing.yml reflecting sys.path additions ( #869 )
...
* Update ci-testing.yml
* Update yolo.py
* Update ci-testing.yml
* Update export.py
* Update yolo.py
2020-10-04 18:50:32 +02:00
PetrDvoracek
d55c326dda
export.py folder sys.path addition ( #1052 )
2020-10-04 17:42:05 +02:00
Glenn Jocher
4fce0096fa
model.add_nms() method
2020-09-20 16:31:40 -07:00
Glenn Jocher
2f77cf33f6
.fuse() additional error checking
2020-09-20 12:04:20 -07:00
Glenn Jocher
c4cb78570c
add NMS to pretrained pytorch hub models
2020-09-18 18:17:11 -07:00
Glenn Jocher
6062bb7dee
print self.yaml['nc'] override update
2020-09-11 17:44:11 -07:00
tomguluson92
35dc935504
remove `nn.Conv2d` in default settings ( #949 )
2020-09-11 16:35:48 -07:00
Glenn Jocher
d68179953d
Update export.py
2020-09-02 13:23:29 -07:00
Glenn Jocher
1e7674d4dd
Update export.py --img-size comment
2020-08-25 23:07:22 -07:00
Glenn Jocher
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
2020-08-24 21:59:26 -07:00
Glenn Jocher
4d7f222f73
Update export.py with v3.0 Hardswish() support
2020-08-24 21:47:49 -07:00
Glenn Jocher
a8751e50de
model.fuse() fix for export.py ( #827 )
2020-08-24 19:27:54 -07:00
Glenn Jocher
d0f98c0095
Update yolo.py with model_info(verbose=False)
2020-08-21 16:20:45 -07:00
Glenn Jocher
b42e8a531b
optimize imports
2020-08-20 21:17:40 -07:00
Glenn Jocher
916d4aad9a
v3.0 Release ( #725 )
...
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
2020-08-13 14:25:05 -07:00
NanoCode012
0892c44bc4
Fix Logging ( #719 )
...
* Add logging setup
* Fix fusing layers message
* Fix logging does not have end
* Add logging
* Change logging to use logger
* Update yolo.py
I tried this in a cloned branch, and everything seems to work fine
* Update yolo.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 14:18:19 -07:00
NanoCode012
4949401a94
Fix redundant outputs via Logging in DDP training ( #500 )
...
* Change print to logging
* Clean function set_logging
* Add line spacing
* Change leftover prints to log
* Fix scanning labels output
* Fix rank naming
* Change leftover print to logging
* Reorganized DDP variables
* Fix type error
* Make quotes consistent
* Fix spelling
* Clean function call
* Add line spacing
* Update datasets.py
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 11:18:45 -07:00
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
Glenn Jocher
d989bc9260
remove NBSP
2020-08-02 14:23:05 -07:00
Glenn Jocher
2268f9c44f
update *.yaml output layer comments
2020-07-30 11:31:47 -07:00
Liu Changyu
adaf1d8ac1
Update models/hub *.yaml files for v2.0 ( #546 )
...
* fix #543 issue
* Update yolov3-spp.yaml
* Update yolov5-fpn.yaml
* Update yolov5-panet.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-29 14:02:07 -07:00
Glenn Jocher
43a616a955
PyTorch 1.6.0 compatability updates
2020-07-28 15:38:56 -07:00
Glenn Jocher
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
...
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
2020-07-24 11:42:23 -07:00
Glenn Jocher
9da56b62dd
v2.0 Release ( #491 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-23 15:34:23 -07:00
Glenn Jocher
a97c3f94ec
update common.py Classify()
2020-07-16 23:59:51 -07:00
Glenn Jocher
5387d4747d
update common.py add Classify()
2020-07-16 17:18:41 -07:00
Glenn Jocher
2b1892430f
yolo.py bug fix
2020-07-13 15:21:19 -07:00
Glenn Jocher
02445d176d
improved model.yaml source tracking
2020-07-13 14:35:47 -07:00
Glenn Jocher
9006b85d21
update export.py torchscript file extension
2020-07-10 11:56:01 -07:00
Glenn Jocher
16f6834486
update train.py and experimental.py
2020-07-08 14:23:34 -07:00
Laurentiu Diaconu
a9918fbf2b
updated coreml conversion to do pixel scaling between 0.0 and 1.0
2020-07-08 09:43:33 +03:00
Glenn Jocher
e8cf24b6c8
Initial model ensemble capability #318
2020-07-07 15:40:50 -07:00
Glenn Jocher
04bdbe4104
fuse update
2020-07-05 23:16:50 -07:00
Glenn Jocher
5ba1de0cdc
update experimental.py with Ensemble() module
2020-07-05 15:02:56 -07:00
Glenn Jocher
38f5c1ad1d
pruning and sparsity initial commit
2020-07-05 13:41:21 -07:00
Glenn Jocher
997ba7b346
import yaml in yolo.py
2020-07-05 12:50:04 -07:00
Glenn Jocher
762b06de4b
Merge remote-tracking branch 'origin/master'
2020-07-04 23:07:32 -07:00
Glenn Jocher
4e2d24602d
update yolo.py
2020-07-04 23:07:26 -07:00
Glenn Jocher
bb3c346916
model.yaml nc inherited from dataset.yaml
2020-07-04 17:51:54 -07:00
Glenn Jocher
659ad748c8
update get_voc.sh
2020-07-04 17:13:43 -07:00
Glenn Jocher
7bd7b2c090
update export.py
2020-07-03 20:05:50 -07:00
Glenn Jocher
655895a838
module updates
2020-07-03 13:46:12 -07:00
Glenn Jocher
a9d20eba3e
export.py update
2020-07-03 11:50:59 -07:00
Glenn Jocher
569fbc63c8
export.py update
2020-07-03 11:29:53 -07:00
Glenn Jocher
a03adb5ab8
Merge remote-tracking branch 'origin/master'
2020-07-01 19:16:06 -07:00
Glenn Jocher
1fca7a7f24
autopad() update in common.py
2020-07-01 19:15:59 -07:00
Glenn Jocher
a62a1c2c67
export.py update
2020-07-01 16:14:49 -07:00
Glenn Jocher
f1d67f4110
update export.py
2020-07-01 15:46:15 -07:00
Glenn Jocher
b5659d1195
module updates
2020-07-01 11:44:49 -07:00
Glenn Jocher
615d6d0cfa
FROM nvcr.io/nvidia/pytorch:20.06-py3
2020-06-29 14:00:13 -07:00
Glenn Jocher
e8bf385795
Merge pull request #223 from ArmyOfRobots/master
...
Adding torchscript export
2020-06-29 13:39:53 -07:00
Glenn Jocher
0bc80e1e57
onnx_export.py bug fix #224
2020-06-28 17:07:03 -07:00
Jake Poznanski
685091a3fc
Adding torchscript export
2020-06-28 13:17:05 -07:00
Glenn Jocher
9a9c4f1259
update yolo.py default --cfg
2020-06-24 18:00:43 -07:00
Glenn Jocher
e071b8dd3a
update models/common.py for Conv() flexible padding
2020-06-24 13:12:23 -07:00
Glenn Jocher
0825cb7fd8
create models/hub
2020-06-24 11:58:23 -07:00
Glenn Jocher
b1b3634145
update onnx_export.py #134
2020-06-24 11:00:03 -07:00
Glenn Jocher
8699c319ca
update yolo.py
2020-06-24 10:42:01 -07:00
Glenn Jocher
53294f28b4
update yolo.py
2020-06-23 10:53:42 -07:00
Glenn Jocher
364fcfd7db
PANet update
2020-06-22 14:27:17 -07:00
Glenn Jocher
fc171e2678
check_anchor_order() update
2020-06-21 14:08:59 -07:00
Glenn Jocher
e249792c2f
update onnx_export.py usage
2020-06-19 10:15:08 -07:00
Glenn Jocher
80b82e8bde
update yolo.py
2020-06-18 19:33:02 -07:00
Glenn Jocher
e1e3399286
ONNX export bug fix #93
2020-06-16 09:59:42 -07:00
ashing
62eda5e9c2
add "__init__.py" file in <models> folder
2020-06-15 14:37:07 +08:00
Glenn Jocher
4ba47b1b1b
model.yaml updates
2020-06-12 15:41:46 -07:00
Glenn Jocher
1e2cb6b480
ONNX export explicit cpu map_location
2020-06-12 13:54:00 -07:00
Glenn Jocher
8f17a96ff0
cleanup
2020-06-11 21:55:22 -07:00
Glenn Jocher
8f1755290c
ONNX export update
2020-06-11 21:54:01 -07:00
Glenn Jocher
3a5c5328c5
ONNX, BCEBlurWithLogitsLoss, plot_study updates
2020-06-11 14:44:09 -07:00
Glenn Jocher
a814720403
PyTorch Hub updates
2020-06-10 19:11:11 -07:00
Glenn Jocher
c14368d768
new CSP model release
2020-06-09 14:31:04 -07:00
Glenn Jocher
854ccb9a74
_print_weights()
2020-06-09 10:51:47 -07:00
Glenn Jocher
5bee686649
model definition update
2020-06-08 23:26:56 -07:00
Glenn Jocher
715cb08b10
module updates
2020-06-08 22:15:16 -07:00
Glenn Jocher
ff02ae0869
module updates
2020-06-08 22:13:01 -07:00
Glenn Jocher
12b0c046d5
model fusion and onnx export
2020-06-07 13:42:33 -07:00
Glenn Jocher
b810b21299
augmented inference
2020-06-05 21:14:15 -07:00
Glenn Jocher
5c470d235e
multi-gpu test bug fix #15
2020-06-05 20:36:29 -07:00
Glenn Jocher
67d5e65c77
profiling update
2020-06-05 20:22:39 -07:00
Glenn Jocher
2703ac73e8
profiling update
2020-06-05 20:22:17 -07:00
Glenn Jocher
4e32b6029a
update
2020-06-05 20:03:21 -07:00
Glenn Jocher
dbdee3a4a3
multi-gpu test bug fix #15
2020-06-04 15:23:03 -07:00
Glenn Jocher
a2336088f0
updates
2020-06-04 14:25:10 -07:00
Glenn Jocher
17e4926cdf
updates
2020-06-02 23:40:14 -07:00
Glenn Jocher
ee8988b8a2
datasets.py self.batch_shapes fix
2020-06-01 18:11:18 -07:00
Glenn Jocher
7a00a5eff2
onnx_export.py
2020-06-01 14:05:01 -07:00
Glenn Jocher
df7988d8d0
onnx_export.py
2020-06-01 13:53:47 -07:00
Glenn Jocher
b10609fe6b
updates
2020-05-31 15:07:04 -07:00
Glenn Jocher
f129cdda9f
updates
2020-05-30 11:03:50 -07:00
Glenn Jocher
1b97392801
updates
2020-05-30 10:46:35 -07:00
Glenn Jocher
1e84a23f38
initial commit
2020-05-29 17:04:54 -07:00