75 Commits (41523e2c91f31fdaee3b05c8d602acd9976080cd)

Author SHA1 Message Date
Jirka Borovec d5b6416c87
Explicit Imports (#498)
5 years ago
Glenn Jocher d989bc9260 remove NBSP
5 years ago
Glenn Jocher 2268f9c44f update *.yaml output layer comments
5 years ago
Liu Changyu adaf1d8ac1
Update models/hub *.yaml files for v2.0 (#546)
5 years ago
Glenn Jocher 43a616a955 PyTorch 1.6.0 compatability updates
5 years ago
Glenn Jocher 1d17b9af0f
update yolo.py TTA flexibility and extensibility (#506)
5 years ago
Glenn Jocher 9da56b62dd
v2.0 Release (#491)
5 years ago
Glenn Jocher a97c3f94ec update common.py Classify()
5 years ago
Glenn Jocher 5387d4747d update common.py add Classify()
5 years ago
Glenn Jocher 2b1892430f yolo.py bug fix
5 years ago
Glenn Jocher 02445d176d improved model.yaml source tracking
5 years ago
Glenn Jocher 9006b85d21 update export.py torchscript file extension
5 years ago
Glenn Jocher 16f6834486 update train.py and experimental.py
5 years ago
Laurentiu Diaconu a9918fbf2b updated coreml conversion to do pixel scaling between 0.0 and 1.0
5 years ago
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318
5 years ago
Glenn Jocher 04bdbe4104 fuse update
5 years ago
Glenn Jocher 5ba1de0cdc update experimental.py with Ensemble() module
5 years ago
Glenn Jocher 38f5c1ad1d pruning and sparsity initial commit
5 years ago
Glenn Jocher 997ba7b346 import yaml in yolo.py
5 years ago
Glenn Jocher 762b06de4b Merge remote-tracking branch 'origin/master'
5 years ago
Glenn Jocher 4e2d24602d update yolo.py
5 years ago
Glenn Jocher bb3c346916 model.yaml nc inherited from dataset.yaml
5 years ago
Glenn Jocher 659ad748c8 update get_voc.sh
5 years ago
Glenn Jocher 7bd7b2c090 update export.py
5 years ago
Glenn Jocher 655895a838 module updates
5 years ago
Glenn Jocher a9d20eba3e export.py update
5 years ago
Glenn Jocher 569fbc63c8 export.py update
5 years ago
Glenn Jocher a03adb5ab8 Merge remote-tracking branch 'origin/master'
5 years ago
Glenn Jocher 1fca7a7f24 autopad() update in common.py
5 years ago
Glenn Jocher a62a1c2c67 export.py update
5 years ago
Glenn Jocher f1d67f4110 update export.py
5 years ago
Glenn Jocher b5659d1195 module updates
5 years ago
Glenn Jocher 615d6d0cfa FROM nvcr.io/nvidia/pytorch:20.06-py3
5 years ago
Glenn Jocher e8bf385795
Merge pull request #223 from ArmyOfRobots/master
5 years ago
Glenn Jocher 0bc80e1e57 onnx_export.py bug fix #224
5 years ago
Jake Poznanski 685091a3fc Adding torchscript export
5 years ago
Glenn Jocher 9a9c4f1259 update yolo.py default --cfg
5 years ago
Glenn Jocher e071b8dd3a update models/common.py for Conv() flexible padding
5 years ago
Glenn Jocher 0825cb7fd8 create models/hub
5 years ago
Glenn Jocher b1b3634145 update onnx_export.py #134
5 years ago
Glenn Jocher 8699c319ca update yolo.py
5 years ago
Glenn Jocher 53294f28b4 update yolo.py
5 years ago
Glenn Jocher 364fcfd7db PANet update
5 years ago
Glenn Jocher fc171e2678 check_anchor_order() update
5 years ago
Glenn Jocher e249792c2f update onnx_export.py usage
5 years ago
Glenn Jocher 80b82e8bde update yolo.py
5 years ago
Glenn Jocher e1e3399286 ONNX export bug fix #93
5 years ago
ashing 62eda5e9c2 add "__init__.py" file in <models> folder
5 years ago
Glenn Jocher 4ba47b1b1b model.yaml updates
5 years ago
Glenn Jocher 1e2cb6b480 ONNX export explicit cpu map_location
5 years ago