1e7674d4dd
Update export.py --img-size comment
2020-08-25 23:07:22 -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
a8751e50de
model.fuse() fix for export.py ( #827 )
2020-08-24 19:27:54 -07:00
d0f98c0095
Update yolo.py with model_info(verbose=False)
2020-08-21 16:20:45 -07:00
b42e8a531b
optimize imports
2020-08-20 21:17:40 -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
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
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
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
d989bc9260
remove NBSP
2020-08-02 14:23:05 -07:00
2268f9c44f
update *.yaml output layer comments
2020-07-30 11:31:47 -07:00
adaf1d8ac1
Update models/hub *.yaml files for v2.0 ( #546 )
...
* fix #543 issue
* Update yolov3-spp.yaml
* Update yolov5-fpn.yaml
* Update yolov5-panet.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-29 14:02:07 -07:00
43a616a955
PyTorch 1.6.0 compatability updates
2020-07-28 15:38:56 -07:00
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
...
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
2020-07-24 11:42:23 -07:00
9da56b62dd
v2.0 Release ( #491 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-23 15:34:23 -07:00
a97c3f94ec
update common.py Classify()
2020-07-16 23:59:51 -07:00
5387d4747d
update common.py add Classify()
2020-07-16 17:18:41 -07:00
2b1892430f
yolo.py bug fix
2020-07-13 15:21:19 -07:00
02445d176d
improved model.yaml source tracking
2020-07-13 14:35:47 -07:00
9006b85d21
update export.py torchscript file extension
2020-07-10 11:56:01 -07:00
16f6834486
update train.py and experimental.py
2020-07-08 14:23:34 -07:00
a9918fbf2b
updated coreml conversion to do pixel scaling between 0.0 and 1.0
2020-07-08 09:43:33 +03:00
e8cf24b6c8
Initial model ensemble capability #318
2020-07-07 15:40:50 -07:00
04bdbe4104
fuse update
2020-07-05 23:16:50 -07:00
5ba1de0cdc
update experimental.py with Ensemble() module
2020-07-05 15:02:56 -07:00
38f5c1ad1d
pruning and sparsity initial commit
2020-07-05 13:41:21 -07:00
997ba7b346
import yaml in yolo.py
2020-07-05 12:50:04 -07:00
762b06de4b
Merge remote-tracking branch 'origin/master'
2020-07-04 23:07:32 -07:00
4e2d24602d
update yolo.py
2020-07-04 23:07:26 -07:00
bb3c346916
model.yaml nc inherited from dataset.yaml
2020-07-04 17:51:54 -07:00
659ad748c8
update get_voc.sh
2020-07-04 17:13:43 -07:00
7bd7b2c090
update export.py
2020-07-03 20:05:50 -07:00
655895a838
module updates
2020-07-03 13:46:12 -07:00
a9d20eba3e
export.py update
2020-07-03 11:50:59 -07:00
569fbc63c8
export.py update
2020-07-03 11:29:53 -07:00
a03adb5ab8
Merge remote-tracking branch 'origin/master'
2020-07-01 19:16:06 -07:00
1fca7a7f24
autopad() update in common.py
2020-07-01 19:15:59 -07:00
a62a1c2c67
export.py update
2020-07-01 16:14:49 -07:00
f1d67f4110
update export.py
2020-07-01 15:46:15 -07:00
b5659d1195
module updates
2020-07-01 11:44:49 -07:00
615d6d0cfa
FROM nvcr.io/nvidia/pytorch:20.06-py3
2020-06-29 14:00:13 -07:00
e8bf385795
Merge pull request #223 from ArmyOfRobots/master
...
Adding torchscript export
2020-06-29 13:39:53 -07:00
0bc80e1e57
onnx_export.py bug fix #224
2020-06-28 17:07:03 -07:00
685091a3fc
Adding torchscript export
2020-06-28 13:17:05 -07:00
9a9c4f1259
update yolo.py default --cfg
2020-06-24 18:00:43 -07:00
e071b8dd3a
update models/common.py for Conv() flexible padding
2020-06-24 13:12:23 -07:00
0825cb7fd8
create models/hub
2020-06-24 11:58:23 -07:00
b1b3634145
update onnx_export.py #134
2020-06-24 11:00:03 -07:00
8699c319ca
update yolo.py
2020-06-24 10:42:01 -07:00
Glenn Jocher
NanoCode012
Jirka Borovec
Liu Changyu
Laurentiu Diaconu
Jake Poznanski