Glenn Jocher
85b6fdda0f
FROM nvcr.io/nvidia/pytorch:21.03-py3
Issues with 21.04
3 years ago
Glenn Jocher
4c7d686d7e
FROM nvcr.io/nvidia/pytorch:21.04-py3 ( #3030 )
3 years ago
Glenn Jocher
8cab907f60
download() ThreadPool update ( #3027 )
* download() ThreadPool update
* update train image count
* cid + 1
3 years ago
Glenn Jocher
5189b3addb
Objects365 Update
3 years ago
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>
3 years ago
Glenn Jocher
41cc7caee6
Add NMS threshold checks ( #3020 )
* Add NMS threshold checks
* fix
3 years ago
Glenn Jocher
fcd57028d2
Add is_colab() function ( #3018 )
3 years ago
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>
3 years ago
Glenn Jocher
31ee54c28c
Hubconf.py bug fix ( #3007 )
3 years ago
Glenn Jocher
7c89c829e3
cleanup
3 years ago
Glenn Jocher
c1a44ed9c5
Update hubconf.py for unified loading ( #3005 )
3 years ago
Glenn Jocher
37eaffec7d
Curl update ( #3004 )
* Curl update
* Curl update
3 years ago
Glenn Jocher
4d7bca7648
Add unzip flag to download() ( #3002 )
3 years ago
Glenn Jocher
f55730ee9c
Objects365 bug fix
3 years ago
Glenn Jocher
54652fe3ff
Objects365 update
3 years ago
Ayush Chaurasia
801b469878
Improve performance of dataset Logger ( #2943 )
* Improve performance of Dataset Logger
* Fix scaling bug
3 years ago
Glenn Jocher
d08575ee5e
PyTorch Hub load directly when possible ( #2986 )
3 years ago
Glenn Jocher
9b91db6d1a
Created using Colaboratory
3 years ago
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>
3 years ago
Glenn Jocher
955eea8b96
Make cache saving optional ( #2977 )
3 years ago
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list ( #2974 )
3 years ago
Ferdinand Loesch
dbce1bc54c
Objects365 Dataset ( #2932 )
* add object365
* ADD CONVERSION SCRIPT
* fix transcript
* Reformat and simplify
* spelling
* Update get_objects365.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
33712d6dd0
Global Wheat Detection 2020 Dataset Auto-Download ( #2968 )
* Create GlobalWheat2020.yaml
* Update and rename visdrone.yaml to VisDrone.yaml
* Update GlobalWheat2020.yaml
3 years ago
Glenn Jocher
2c7c075fb1
Update restapi.py ( #2967 )
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 )
* Add yolov5/ to sys.path() for *.py subdir exec
* Update export.py
3 years ago
Glenn Jocher
4890499344
Improved yolo.py profiling ( #2940 )
* Improved yolo.py profiling
Improved column order and labelling.
* Update yolo.py
3 years ago
BZFYS
6c1290fe03
Update README.md ( #2934 )
* Update README.md
dependencies:
ImportError: libGL.so.1: cannot open shared object file: No such file or directory
ImportError: libgthread-2.0.so.0: cannot open shared object file: No such file or directory
ImportError: libSM.so.6: cannot open shared object file: No such file or directory
ImportError: libXrender.so.1: cannot open shared object file: No such file or directory
* replace older apt-get with apt
Code commented for now until a better understanding of the issue, and also code is not cross-platform compatible.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
1849916726
Check_requirements() enclosing apostrophe bug fix ( #2929 )
This fixes a bug where the '>' symbol in python package requirements was not running correctly with subprocess.check_output() commands.
3 years ago
Glenn Jocher
9c7bb5a52c
ACON Activation batch-size 1 bug patch ( #2901 )
* ACON Activation batch-size 1 bug path
This is not a great solution to https://github.com/nmaac/acon/issues/4 but it's all I could think of at the moment.
WARNING: YOLOv5 models with MetaAconC() activations are incapable of running inference at batch-size 1 properly due to a known bug in https://github.com/nmaac/acon/issues/4 with no known solution.
* Update activations.py
* Update activations.py
* Update activations.py
* Update activations.py
3 years ago
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>
3 years ago
Glenn Jocher
3665c0f59b
test.py native --single-cls ( #2928 )
3 years ago
albinxavi
aa78069c58
Change default value of hide-conf argument to false ( #2925 )
3 years ago
albinxavi
eae28a93b0
Change default value of hide label argument to False ( #2923 )
3 years ago
Glenn Jocher
de19d396e7
Update visdrone.yaml ( #2921 )
3 years ago
Glenn Jocher
45632b2704
Update download() for tar.gz files ( #2919 )
* Update download() for tar.gz files
* Update general.py
3 years ago
Glenn Jocher
1b1ab4cca2
Add file_size() function ( #2911 )
* Add file_size() function
* Update export.py
3 years ago
Glenn Jocher
646386ff09
Update export.py for 2 dry runs ( #2910 )
* Update export.py for 2 dry runs
* Update export.py
3 years ago
Glenn Jocher
a2a514dec8
Update export.py ( #2909 )
3 years ago
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).
3 years ago
Maximilian Peters
f662c18507
Add detect.py --hide-conf --hide-labels --line-thickness options ( #2658 )
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for hiding confidence values
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
14d2d2d75f
Update google_utils.py ( #2900 )
3 years ago
Glenn Jocher
cdb678f418
Update yolo.py ( #2899 )
3 years ago
fcakyon
b40dd99167
Explicit opt function arguments ( #2817 )
* more explicit function arguments
* fix typo in detect.py
* revert import order
* revert import order
* remove default value
3 years ago
Glenn Jocher
264d860f8d
ACON activation function ( #2893 )
* ACON Activation Function
## 🚀 Feature
There is a new activation function [ACON (CVPR 2021)](https://arxiv.org/pdf/2009.04759.pdf ) that unifies ReLU and Swish.
ACON is simple but very effective, code is here: https://github.com/nmaac/acon/blob/main/acon.py#L19
![image](https://user-images.githubusercontent.com/5032208/115676962-a38dfe80-a382-11eb-9883-61fa3216e3e6.png )
The improvements are very significant:
![image](https://user-images.githubusercontent.com/5032208/115680180-eac9be80-a385-11eb-9c7a-8643db552c69.png )
## Alternatives
It also has an enhanced version meta-ACON that uses a small network to learn beta explicitly, which may influence the speed a bit.
## Additional context
[Code](https://github.com/nmaac/acon ) and [paper](https://arxiv.org/pdf/2009.04759.pdf ).
* Update activations.py
3 years ago
r-blmnr
0395e39fd0
Uppercase model filenames enabled ( #2890 )
3 years ago
Glenn Jocher
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
3 years ago
Michael Heilig
d48a34dca7
bug fix: switched rows and cols for correct detections in confusion matrix ( #2883 )
3 years ago
JoshSong
5f7d39fede
Cleanup load_image() ( #2871 )
* don't resize up in load_image if augmenting
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3 years ago