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
82ed33a052
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462 )
2020-08-25 11:07:38 -07:00
09402a2174
torch.from_tensor() bug fix
2020-08-25 03:14:17 -07:00
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
2020-08-24 21:59:26 -07:00
4d7f222f73
Update export.py with v3.0 Hardswish() support
2020-08-24 21:47:49 -07:00
8666bc507e
Create evolve.sh for infinite hyp evolution
2020-08-24 20:27:28 -07:00
a8751e50de
model.fuse() fix for export.py ( #827 )
2020-08-24 19:27:54 -07:00
5f07782f56
Update requirements.txt for coremltools kmeans quantization
2020-08-22 20:50:53 -07:00
71209a6099
exportable Hardswish() implementation
2020-08-22 16:59:31 -07:00
fd71fe8451
Update hyp.finetune.yaml
2020-08-22 16:18:53 -07:00
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
2020-08-22 15:47:39 -07:00
9dbaa1bbbe
Update plot_results()
2020-08-22 15:32:05 -07:00
83dc540b1d
remove ema.ema hasattr(ema, 'module') check
2020-08-22 15:18:39 -07:00
1309400e6b
Update hyp.finetune.yaml on VOC results
2020-08-22 15:05:47 -07:00
fb05681a04
Update requirements.txt to coremltools==4.0b3
2020-08-22 14:22:15 -07:00
d0f98c0095
Update yolo.py with model_info(verbose=False)
2020-08-21 16:20:45 -07:00
b9851222bf
Merge remote-tracking branch 'origin/master'
2020-08-21 15:03:09 -07:00
c171e45cda
Update test.py with set() ( #806 )
2020-08-21 15:03:03 -07:00
ea5e97fb9b
FP16 NMS ( #814 )
2020-08-21 14:44:25 -07:00
b42e8a531b
optimize imports
2020-08-20 21:17:40 -07:00
4447f4b937
--resume to same runs/exp directory ( #765 )
...
* initial commit
* add weight backup dir on resume
2020-08-20 18:24:33 -07:00
5e0b90de8f
CIoU nan bug fix ( #736 )
2020-08-20 10:37:55 -07:00
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
da35727c68
--save-txt bug fix ( #763 )
2020-08-17 17:14:07 -07:00
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
26c3b11fdc
Update torch_utils.py ( #751 )
2020-08-16 13:58:59 -07:00
2077d78880
update Dockerfile with libgl1-mesa-glx
2020-08-15 09:53:22 -07:00
f5da528d28
reformat code
2020-08-14 11:53:44 -07:00
b35f76e3d0
update .dockerignore
2020-08-14 11:29:15 -07:00
fe80bed695
Update stale.yml
2020-08-14 10:58:33 -07:00
7b2b52193d
update requirements.txt
2020-08-14 10:14:48 -07:00
5991d144c1
Update README.md
2020-08-14 00:04:12 -07:00
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
06bdd3955c
Update README.md
2020-08-13 14:31:08 -07:00
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
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
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
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
a925f283a7
max workers for dataloader ( #722 )
2020-08-12 13:57:36 -07:00
7875f4c1fb
reverse plotting low to high confidence
2020-08-12 13:50:16 -07:00
d2da523053
Merge remote-tracking branch 'origin/master'
2020-08-11 20:11:58 -07:00
9f482cbcb8
utils.general comment updates/bug fixes
2020-08-11 20:11:44 -07:00
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
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
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
e71fd0ec0b
Model freeze capability ( #679 )
2020-08-10 22:49:43 -07:00
9ae868364a
the value should be a list to unpacking ( #688 )
2020-08-10 11:23:48 -07:00
8e5c66579b
update train.py remove save_json final_epoch
2020-08-09 21:24:40 -07:00
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
3d8ed0a76b
Fix missing model.stride in DP and DDP mode ( #683 )
2020-08-09 11:01:36 -07:00
Glenn Jocher
NanoCode012
Francisco Ingham
Quan Lin
tkianai
Hatovix
Marc