Glenn Jocher
|
3e04d20c7d
|
Merge remote-tracking branch 'origin/master'
|
2020-08-31 11:07:00 -07:00 |
|
Glenn Jocher
|
f06e2d518c
|
opt.image_weights bug fix (#885)
|
2020-08-31 11:05:37 -07:00 |
NanoCode012
|
1e15aad6f9
|
Add InfiniteDataLoader class (#876)
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
|
2020-08-31 11:01:25 -07:00 |
|
Glenn Jocher
|
69ff781ca5
|
opt.img_weights bug fix (#885)
|
2020-08-31 10:33:07 -07:00 |
|
Glenn Jocher
|
987c226849
|
seaborn plotting correlogram addition
|
2020-08-29 15:13:58 -07:00 |
|
Glenn Jocher
|
08e97a2f88
|
Update hyperparameters to add lrf, anchors
|
2020-08-28 14:58:43 -07:00 |
rafale77
|
9776e70988
|
torch.ops.torchvision.nms (#860)
Don't load the entire torchvision library just for nms when the function is already in the torch library.
|
2020-08-27 11:13:41 -07:00 |
|
Glenn Jocher
|
0c01afc390
|
Update Dockerfile to upgrade pip
|
2020-08-26 14:44:43 -07:00 |
|
Glenn Jocher
|
62e037419f
|
Update Dockerfile FROM nvcr.io/nvidia/pytorch:20.08-py3
|
2020-08-26 00:44:34 -07:00 |
|
Glenn Jocher
|
31d6c389f2
|
Update general.py plot_evolution()
|
2020-08-25 23:08:14 -07:00 |
|
Glenn Jocher
|
1e7674d4dd
|
Update export.py --img-size comment
|
2020-08-25 23:07:22 -07:00 |
|
Glenn Jocher
|
a21bd0687c
|
Update train.py forward simplification
|
2020-08-25 13:48:03 -07:00 |
Willie Maddox
|
455f7b8f76
|
Update general.py (#823)
Fixes #822
`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
|
2020-08-25 13:01:17 -07:00 |
|
Glenn Jocher
|
5a7d79fbe6
|
Update general.py bbox_iou() with eps improvements (#736)
eps reduced to 1e-9 for additional precaution.
|
2020-08-25 12:56:39 -07:00 |
|
Glenn Jocher
|
82ed33a052
|
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462)
|
2020-08-25 11:07:38 -07:00 |
|
Glenn Jocher
|
09402a2174
|
torch.from_tensor() bug fix
|
2020-08-25 03:14:17 -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
|
8666bc507e
|
Create evolve.sh for infinite hyp evolution
|
2020-08-24 20:27:28 -07:00 |
|
Glenn Jocher
|
a8751e50de
|
model.fuse() fix for export.py (#827)
|
2020-08-24 19:27:54 -07:00 |
|
Glenn Jocher
|
5f07782f56
|
Update requirements.txt for coremltools kmeans quantization
|
2020-08-22 20:50:53 -07:00 |
|
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 |