Skip to content

Commit 2328f2b

Browse files
authored
Update README.md
1 parent 40dc14a commit 2328f2b

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

README.md

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ A [PyTorch](http://pytorch.org/) implementation of [Single Shot MultiBox Detecto
33

44
## Authors
55

6-
* [**Max de Groot**](https://github.com/amdegroot)
6+
* [**Max deGroot**](https://github.com/amdegroot)
77
* [**Ellis Brown**](http://github.com/ellisbrown)
88

99
***Note:*** Unfortunately, this is just a hobby of ours and not a full-time job, so we'll do our best to keep things up to date, but no guarantees. That being said, thanks to everyone for your continued help and feedback as it is really appreciated. We will try to address everything as soon as possible.
@@ -78,7 +78,6 @@ python train.py
7878

7979
- Note:
8080
* For training, an NVIDIA GPU is strongly recommended for speed.
81-
* Currently we only support training on v2 (the newest version).
8281
* For instructions on Visdom usage/installation, see the <a href='#installation'>Installation</a> section.
8382
* You can pick-up training from a checkpoint by specifying the path as one of the training parameters (again, see `train.py` for options)
8483

@@ -114,9 +113,9 @@ You can specify the parameters listed in the `eval.py` file by flagging them or
114113
#### Download a pre-trained network
115114
- We are trying to provide PyTorch `state_dicts` (dict of weight tensors) of the latest SSD model definitions trained on different datasets.
116115
- Currently, we provide the following PyTorch models:
117-
* SSD300 v2 trained on VOC0712 (newest PyTorch version)
116+
* SSD300 trained on VOC0712 (newest PyTorch weights)
118117
- https://s3.amazonaws.com/amdegroot-models/ssd300_mAP_77.43_v2.pth
119-
* SSD300 v2 trained on VOC0712 (original Caffe version)
118+
* SSD300 trained on VOC0712 (original Caffe weights)
120119
- https://s3.amazonaws.com/amdegroot-models/ssd_300_VOC0712.pth
121120
- Our goal is to reproduce this table from the [original paper](http://arxiv.org/abs/1512.02325)
122121
<p align="left">
@@ -152,10 +151,9 @@ jupyter notebook
152151
## TODO
153152
We have accumulated the following to-do list, which you can expect to be done in the very near future
154153
- Still to come:
155-
* Train SSD300 with batch norm
156-
* Add support for SSD512 training and testing
157-
* Add support for COCO dataset
158-
* Create a functional model definition for Sergey Zagoruyko's [functional-zoo](https://github.com/szagoruyko/functional-zoo)
154+
* Support for the MS COCO dataset
155+
* Support for SSD512 training and testing
156+
* Support for training on custom datasets
159157

160158

161159
## References

0 commit comments

Comments
 (0)