41 Commits

Author SHA1 Message Date
Glenn Jocher
166a4d590f
v9.1 release (#1658) 2021-01-12 23:05:32 -08:00
Glenn Jocher
61fb2dbd20
Simplify autoshape() post-process (#1603)
* Simplify autoshape() post-process

* cleanup
2020-12-09 07:43:58 -08:00
Glenn Jocher
76807fae71
YOLOv5 Forward Compatibility Update (#1569)
* YOLOv5 forward compatibility update

* add data dir

* ci test yolov3

* update build_targets()

* update build_targets()

* update build_targets()

* update yolov3-spp.yaml

* add yolov3-tiny.yaml

* add yolov3-tiny.yaml

* Update yolov3-tiny.yaml

* thop bug fix

* Detection() device bug fix

* Use torchvision.ops.nms()

* Remove redundant download mirror

* CI tests with yolov3-tiny

* Update README.md

* Synch train and test iou_thresh

* update requirements.txt

* Cat apriori autolabels

* Confusion matrix

* Autosplit

* Autosplit

* Update README.md

* AP no plot

* Update caching

* Update caching

* Caching bug fix

* --image-weights bug fix

* datasets bug fix

* mosaic plots bug fix

* plot_study

* boxes.max()

* boxes.max()

* boxes.max()

* boxes.max()

* boxes.max()

* boxes.max()

* update

* Update README

* Update README

* Update README.md

* Update README.md

* results png

* Update README

* Targets scaling bug fix

* update plot_study

* update plot_study

* update plot_study

* update plot_study

* Targets scaling bug fix

* Finish Readme.md

* Finish Readme.md

* Finish Readme.md

* Update README.md

* Creado con Colaboratory
2020-11-26 20:24:00 +01:00
e96031413
4d49957f5a
Update requirements.txt (#1481)
* Update requirements.txt

I found that if we would like to calculate FLOPS in this project, we must install thop.
but there's no thop package inside the requirements.txt

https://github.com/ultralytics/yolov3/blob/master/utils/torch_utils.py#L108

* Update requirements.txt

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-09-10 12:00:53 -07:00
Glenn Jocher
2a74d1fd7d update requirements.txt 2020-08-08 12:57:41 -07:00
Glenn Jocher
ee82e3db5d update requirements.txt (#1431) 2020-08-03 19:37:38 -07:00
Glenn Jocher
bdf546150d
Update requirements.txt #1339 2020-07-08 21:33:10 -07:00
Jason Nataprawira
e1fb453079
Update requirements.txt (#1339)
Add torchvision
2020-06-25 06:09:41 -07:00
Glenn Jocher
3f27ef1253 pycocotools and numpy 1.17 fix for #1182 2020-05-15 20:50:58 -07:00
Josh Veitch-Michaelis
18702c9608
add tensorboard to requirements (#1100)
In a clean environment running training fails if tensorboard is not installed e.g.

```
Traceback (most recent call last):
  File "/home/josh/miniconda3/envs/ultralytics/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py", line 2, in <module>
    from tensorboard.summary.writer.record_writer import RecordWriter  # noqa F401
ModuleNotFoundError: No module named 'tensorboard'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "train.py", line 6, in <module>
    from torch.utils.tensorboard import SummaryWriter
  File "/home/josh/miniconda3/envs/ultralytics/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py", line 4, in <module>
    raise ImportError('TensorBoard logging requires TensorBoard with Python summary writer installed. '
ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above.
```
2020-04-27 09:21:34 -07:00
Glenn Jocher
55757421de remove future 2020-04-26 14:09:12 -07:00
Glenn Jocher
3bf0cb9c60 remove tb-nightly 2020-04-26 12:41:15 -07:00
Glenn Jocher
aa854ecaa9 torch >= 1.5 2020-04-22 17:54:56 -07:00
Glenn Jocher
ee4f7a324d updates 2020-01-23 12:24:52 -08:00
Glenn Jocher
85fbb903f7 updates 2020-01-18 11:52:26 -08:00
Glenn Jocher
75933e93a1 updates 2020-01-16 09:47:33 -08:00
Glenn Jocher
c693219e57 updates 2019-12-22 08:12:23 -08:00
Glenn Jocher
df1be4c748 updates 2019-12-14 14:49:18 -08:00
Glenn Jocher
3953d5c8b0 updates 2019-12-08 16:21:57 -08:00
Glenn Jocher
5e747f8da9 updates 2019-12-06 14:13:07 -08:00
Glenn Jocher
cadd2f75ff updates 2019-12-02 16:46:15 -08:00
Glenn Jocher
55a6b05228 updates 2019-11-23 09:35:11 -10:00
Glenn Jocher
bdf11ffdf1 updates 2019-11-23 09:25:21 -10:00
Glenn Jocher
50a93e141c updates
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2019-09-04 10:05:42 +02:00
Glenn Jocher
c27d8d69a6 updates 2019-08-29 20:19:05 +02:00
Glenn Jocher
48af6d136f updates 2019-08-14 15:54:40 +02:00
LukeAI
891b85490e Update requirements.txt (#444) 2019-08-12 12:38:36 +02:00
Glenn Jocher
933f85f632 tensorboard updates 2019-08-09 16:37:19 +02:00
Marc
22f75469ac Tensorboard support (#435) 2019-08-08 22:30:34 +02:00
Glenn Jocher
65d70fca78 updates 2019-07-30 13:45:44 +02:00
Glenn Jocher
01aaf2c11c updates 2019-07-30 12:39:17 +02:00
Glenn Jocher
666d772fa7 updates 2019-07-25 13:37:01 +02:00
Glenn Jocher
9c227dd2b4 updates 2019-07-09 14:18:19 +02:00
Glenn Jocher
b0aadff56f
Update requirements.txt 2019-07-05 12:34:17 +02:00
Glenn Jocher
1a757524bf add *.jpeg support 2019-05-09 11:23:36 +02:00
Glenn Jocher
bd440fa0c3 updates 2019-03-25 18:35:39 +01:00
Glenn Jocher
cd51e1137b
Add collate_fn() to DataLoader (#163)
Multi-GPU update with custom collate function to allow variable size target vector per image without needing to pad targets.
2019-03-25 14:59:38 +01:00
Glenn Jocher
d6abdaf8d0 updates 2019-02-08 17:17:48 +01:00
Glenn Jocher
178e1a346b updates 2019-01-05 17:23:17 +02:00
Glenn Jocher
c591936446 updates 2018-12-11 21:49:56 +01:00
Glenn Jocher
c3731591af Initial commit 2018-08-26 10:51:39 +02:00