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

588 Commits (71209a60997ad71bf87bc95304e16af2a766758c)
 

Author SHA1 Message Date
Glenn Jocher 71209a6099 exportable Hardswish() implementation 2020-08-22 16:59:31 -07:00
Glenn Jocher fd71fe8451 Update hyp.finetune.yaml 2020-08-22 16:18:53 -07:00
Glenn Jocher f7e075fe93 gsutil cp yaml when evolving to GCP bucket 2020-08-22 15:47:39 -07:00
Glenn Jocher 9dbaa1bbbe Update plot_results() 2020-08-22 15:32:05 -07:00
Glenn Jocher 83dc540b1d remove ema.ema hasattr(ema, 'module') check 2020-08-22 15:18:39 -07:00
Glenn Jocher 1309400e6b Update hyp.finetune.yaml on VOC results 2020-08-22 15:05:47 -07:00
Glenn Jocher fb05681a04 Update requirements.txt to coremltools==4.0b3 2020-08-22 14:22:15 -07:00
Glenn Jocher d0f98c0095 Update yolo.py with model_info(verbose=False) 2020-08-21 16:20:45 -07:00
Glenn Jocher b9851222bf Merge remote-tracking branch 'origin/master' 2020-08-21 15:03:09 -07:00
Glenn Jocher c171e45cda Update test.py with set() (#806) 2020-08-21 15:03:03 -07:00
Glenn Jocher ea5e97fb9b
FP16 NMS (#814) 2020-08-21 14:44:25 -07:00
Glenn Jocher b42e8a531b optimize imports 2020-08-20 21:17:40 -07:00
Glenn Jocher 4447f4b937
--resume to same runs/exp directory (#765) 
* initial commit

* add weight backup dir on resume
 2020-08-20 18:24:33 -07:00
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