mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
575 Commits
10 Branches
10 Tags
584 MiB
Commit Graph

575 Commits (5e0b90de8f7782b3803fa2886bb824c2336358d0)
 

Author SHA1 Message Date
Glenn Jocher 5e0b90de8f CIoU nan bug fix (#736) 2020-08-20 10:37:55 -07:00
NanoCode012 fb4fc8cd02
Fix ema attribute error in DDP mode (#775) 
* Fix ema error in DDP mode

* Update train.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-18 09:52:21 -07:00
Glenn Jocher da35727c68
--save-txt bug fix (#763) 2020-08-17 17:14:07 -07:00
Glenn Jocher ebafd1ead5
single command --resume (#756) 
* single command --resume

* else check files, remove TODO

* argparse.Namespace()

* tensorboard lr

* bug fix in get_latest_run()
 2020-08-17 16:28:43 -07:00
Francisco Ingham 26c3b11fdc
Update torch_utils.py (#751) 2020-08-16 13:58:59 -07:00
Glenn Jocher 2077d78880 update Dockerfile with libgl1-mesa-glx 2020-08-15 09:53:22 -07:00
Glenn Jocher f5da528d28 reformat code 2020-08-14 11:53:44 -07:00
Glenn Jocher b35f76e3d0 update .dockerignore 2020-08-14 11:29:15 -07:00
Glenn Jocher fe80bed695
Update stale.yml 2020-08-14 10:58:33 -07:00
Glenn Jocher 7b2b52193d update requirements.txt 2020-08-14 10:14:48 -07:00
Glenn Jocher 5991d144c1
Update README.md 2020-08-14 00:04:12 -07:00
Quan Lin 66744a0df1
Update utils/google_utils.py (#733) 
To download weights from v3.0 instead of v2.0
 2020-08-13 18:44:14 -07:00
Glenn Jocher 06bdd3955c
Update README.md 2020-08-13 14:31:08 -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
tkianai 1ddf692bf8
Update readme to remove apex dependency (#727) 
* the value should be a list to unpacking

* filter the None labels

* update readme to remove apex dependency
 2020-08-12 21:18:29 -07:00
Hatovix 56c2c344ff
Fix list paths (#721) 
* Add list paths on check_dataset

* missing raise statement

* Update general.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-12 18:53:38 -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
Marc a925f283a7
max workers for dataloader (#722) 2020-08-12 13:57:36 -07:00
Glenn Jocher 7875f4c1fb reverse plotting low to high confidence 2020-08-12 13:50:16 -07:00
Glenn Jocher d2da523053 Merge remote-tracking branch 'origin/master' 2020-08-11 20:11:58 -07:00
Glenn Jocher 9f482cbcb8 utils.general comment updates/bug fixes 2020-08-11 20:11:44 -07:00
Marc 1f92422e20
Github release assets model autodownload (#711) 
* assets autodownload

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

* Update google_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-11 17:17:24 -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
tkianai 542833c997
Filter the None labels (#705) 
* the value should be a list to unpacking

* filter the None labels
 2020-08-11 11:15:48 -07:00
Glenn Jocher e71fd0ec0b Model freeze capability (#679) 2020-08-10 22:49:43 -07:00
tkianai 9ae868364a
the value should be a list to unpacking (#688) 2020-08-10 11:23:48 -07:00
Glenn Jocher 8e5c66579b update train.py remove save_json final_epoch 2020-08-09 21:24:40 -07:00
Glenn Jocher 41523e2c91
Dataset autodownload feature addition (#685) 
* initial commit

* move download scripts into data/scripts

* new check_dataset() function in general.py

* move check_dataset() out of with context

* Update general.py

* DDP update

* Update general.py
 2020-08-09 20:52:57 -07:00
NanoCode012 3d8ed0a76b
Fix missing model.stride in DP and DDP mode (#683) 2020-08-09 11:01:36 -07:00
Glenn Jocher a0ac5adb7b Single-source training update (#680) 2020-08-09 02:27:35 -07:00
Glenn Jocher 3c6e2f7668
Single-source training (#680) 
* Single-source training

* Extract hyperparameters into seperate files

* weight decay scientific notation yaml reader bug fix

* remove import glob

* intersect_dicts() implementation

* 'or' bug fix

* .to(device) bug fix
 2020-08-09 02:12:44 -07:00
NanoCode012 d7cfbc47ab
Fix unrecognized local rank argument (#676) 2020-08-08 16:40:10 -07:00
Glenn Jocher 0e207551aa
update greetings.yml with pull_request_target (#677) 
Should resolve "Resource not accessible by integration" error message.

https://github.com/actions/first-interaction/issues/10#issuecomment-670968624
 2020-08-08 15:35:40 -07:00
Glenn Jocher e742dcf417 update requirements.txt 2020-08-08 12:57:05 -07:00
Glenn Jocher 3fcd365ba3 Fix curl download on Windows (#669) 2020-08-07 12:13:55 -07:00
NanoCode012 fe5b3f8712
Fix curl download on Windows (#669) 
* Update ci-testing.yml

* Fix windows download

* Fix cookie and curl download issue from gdrive

* Revert "Update ci-testing.yml"

This reverts commit 7389d2238d.

* Update google_utils.py

PEP8

* Update google_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-07 12:10:59 -07:00
Glenn Jocher 93684531c6
train.py --logdir argparser addition (#660) 
* train.py --logdir argparser addition

* train.py --logdir argparser addition
 2020-08-06 22:26:38 -07:00
NanoCode012 886b9841c8
Add Multi-Node support for DDP Training (#504) 
* Add support for multi-node DDP

* Remove local_rank confusion

* Fix spacing
 2020-08-06 11:15:24 -07:00
Glenn Jocher 7eaf225d55 zero-target training bug fix (#609) 2020-08-05 13:35:31 -07:00
hopesala d0d3dd1040
Update detect.py (#629) 
* Update detect.py

Update print info for relative path and absolute
otherwise the info below is misleading

Results saved to /workspace//output_path

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-05 12:01:24 -07:00
Glenn Jocher b8f3b1b373
Update ci-testing.yml (#637) 
* Update ci-testing.yml

* Update ci-testing.yml
 2020-08-05 11:42:51 -07:00
lorenzomammana 728efa6576
Fix missing imports (#627) 
* Fix missing imports

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-04 12:22:15 -07:00
Glenn Jocher 941cb2680b Merge remote-tracking branch 'origin/master' 2020-08-03 19:55:57 -07:00
Glenn Jocher 52b626392f update README.md 2020-08-03 19:55:12 -07:00
Wang Xinyu 7ecf09d6da
Update README.md, deploy model with tensorrt (#623) 
* Update README.md, deploy model with tensorrt

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-08-03 19:41:50 -07:00
Glenn Jocher 4ee8adb04b Merge remote-tracking branch 'origin/master' 2020-08-03 17:18:37 -07:00
Glenn Jocher f346da9f2b update activations.py 2020-08-03 17:18:31 -07:00
Glenn Jocher f807e7b867
Update README.md 2020-08-03 11:11:12 -07:00
Glenn Jocher 937b9db977
Update README.md 2020-08-03 11:09:46 -07:00
Glenn Jocher eb99dff9ef import random bug fix (#614) 2020-08-03 10:52:59 -07:00