mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,212 Commits 10 Branches 10 Tags
Commit Graph

1212 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher
b098600763
Update requirements.txt onnx>=1.9.0 (#3143) 2021-05-12 19:56:12 +02:00
Glenn Jocher
f12cef85a3
TorchScript, ONNX, CoreML Export tutorial title (#3142) 2021-05-12 19:49:12 +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
KC-Zhang
bd6f6a784a
Add yolov5/__init__.py (#3127) 2021-05-12 18:50:35 +02:00
Glenn Jocher
7918eed040
Update https://ultralytics.com/images/zidane.jpg (#3140) 2021-05-12 18:30:31 +02:00
Glenn Jocher
c1c7eb023f
Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
Glenn Jocher
52c05707d6
Update Colors() (#3046) 
* Update Colors()

* update colors

* update colors
 2021-05-12 17:41:11 +02:00
Glenn Jocher
abfcf9eb79
Replace print() with logging.info() in trainloader (#3103) 
Might indirectly help #3095 by providing better visibility on source of corruption.
 2021-05-10 17:07:16 +02:00
Glenn Jocher
25f8ab835e
detect.py streaming source --save-crop bug fix (#3102) 
* detect.py streaming source --save-crop bug fix

Possible fix for #3100.

* () parenthesis
 2021-05-10 16:39:37 +02:00
Glenn Jocher
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images (#3101) 2021-05-10 16:04:10 +02:00
Glenn Jocher
60fe54449d
Update train.py (#3099) 2021-05-10 15:26:23 +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
57b0d3a605
Add check_python() (#3088) 
* Add check_python()

Checks python version against minimum version of 3.7.0.

* remove packaging dependency

* refactor import
 2021-05-09 15:33:38 +02:00
Glenn Jocher
91547edec1
Update detect.py (#3087) 
* Update detect.py

* Update detect.py
 2021-05-09 13:01:58 +02:00
Glenn Jocher
251aeafcb1
Update P5 + P6 model ensembling (#3082) 2021-05-08 19:06:12 +02:00
Ayush Chaurasia
d2a17289c9
Explicitly convert artifact path to posix_path (#3067) 
* Explicitly convert artifact path to posix_path

* Remove redudant str() casting
 2021-05-07 19:53:56 +02:00
Glenn Jocher
e97d129db4
Update export.py with --train mode argument (#3066) 2021-05-07 13:10:07 +02:00
kepler62f
f2de1ad2aa
Comment fix (#3058) 
Co-authored-by: kepler62f <>
 2021-05-07 12:28:55 +02:00
Glenn Jocher
3ef3a95cfa
Do not optimize CoreML TorchScript model (#3055) 2021-05-06 17:23:33 +02:00
Jiacong Fang
b8be76f915
Add get_coco128.sh for downloading the coco128 dataset (#3047) 
* Add get_coco128.sh

* Update get_coco128.sh

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-06 13:44:14 +02:00
Ayush Chaurasia
b18ca31ff5
use check_file for --data (#3035) 2021-05-05 14:28:19 +02:00
Glenn Jocher
85b6fdda0f
FROM nvcr.io/nvidia/pytorch:21.03-py3 
Issues with 21.04
 2021-05-05 01:00:12 +02:00
Glenn Jocher
4c7d686d7e
FROM nvcr.io/nvidia/pytorch:21.04-py3 (#3030) 2021-05-05 00:16:35 +02:00
Glenn Jocher
8cab907f60
download() ThreadPool update (#3027) 
* download() ThreadPool update

* update train image count

* cid + 1
 2021-05-04 23:37:31 +02:00
Glenn Jocher
5189b3addb Objects365 Update 2021-05-04 02:25:14 +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
Glenn Jocher
41cc7caee6
Add NMS threshold checks (#3020) 
* Add NMS threshold checks

* fix
 2021-05-03 15:07:20 +02:00
Glenn Jocher
fcd57028d2
Add is_colab() function (#3018) 2021-05-03 12:37:50 +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
Glenn Jocher
31ee54c28c
Hubconf.py bug fix (#3007) 2021-05-01 18:15:17 +02:00
Glenn Jocher
7c89c829e3 cleanup 2021-05-01 17:48:12 +02:00
Glenn Jocher
c1a44ed9c5
Update hubconf.py for unified loading (#3005) 2021-05-01 17:35:02 +02:00
Glenn Jocher
37eaffec7d
Curl update (#3004) 
* Curl update

* Curl update
 2021-05-01 17:29:51 +02:00
Glenn Jocher
4d7bca7648
Add unzip flag to download() (#3002) 2021-05-01 12:37:41 +02:00
Glenn Jocher
f55730ee9c Objects365 bug fix 2021-05-01 01:03:09 +02:00
Glenn Jocher
54652fe3ff Objects365 update 2021-05-01 00:16:58 +02:00
Ayush Chaurasia
801b469878
Improve performance of dataset Logger (#2943) 
* Improve performance of Dataset Logger

* Fix scaling bug
 2021-04-30 21:54:26 +02:00
Glenn Jocher
d08575ee5e
PyTorch Hub load directly when possible (#2986) 2021-04-30 14:59:51 +02:00
Glenn Jocher
9b91db6d1a Created using Colaboratory 2021-04-30 13:15:31 +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
955eea8b96
Make cache saving optional (#2977) 2021-04-30 00:56:44 +02:00
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list (#2974) 2021-04-29 21:16:23 +02:00
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>
 2021-04-29 20:43:33 +02:00
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
 2021-04-28 20:11:02 +02:00
Glenn Jocher
2c7c075fb1
Update restapi.py (#2967) 2021-04-28 17:50:07 +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
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>
 2021-04-26 23:56:25 +02:00
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.
 2021-04-25 23:33:25 +02:00