Skip to content

Commit 1c861f9

Browse files
committed
update readme
1 parent 8c7ac4b commit 1c861f9

File tree

1 file changed

+15
-12
lines changed

1 file changed

+15
-12
lines changed

README.md

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -4,27 +4,30 @@ implementation of YOLOv2.
44
This project is mainly based on [darkflow](https://github.com/thtrieu/darkflow)
55
and [darknet](https://github.com/pjreddie/darknet).
66

7-
For details about YOLO and YOLOv2 please refer to their [project page](https://pjreddie.com/darknet/yolo/)
8-
and the [paper](https://arxiv.org/abs/1612.08242):
9-
YOLO9000: Better, Faster, Stronger by Joseph Redmon and Ali Farhadi.
10-
117
I used a Cython extension for postprocessing and
128
`multiprocessing.Pool` for image preprocessing.
139
Testing an image in VOC2007 costs about 13~20ms.
1410

15-
**NOTE:**
11+
For details about YOLO and YOLOv2 please refer to their [project page](https://pjreddie.com/darknet/yolo/)
12+
and the [paper](https://arxiv.org/abs/1612.08242):
13+
*YOLO9000: Better, Faster, Stronger by Joseph Redmon and Ali Farhadi*.
14+
15+
**NOTE 1:**
1616
This is still an experimental project.
1717
VOC07 test mAP is about 0.71 (trained on VOC07+12 trainval,
1818
reported by [@cory8249](https://github.com/longcw/yolo2-pytorch/issues/23)).
19-
See https://github.com/longcw/yolo2-pytorch/issues/1 and https://github.com/longcw/yolo2-pytorch/issues/23
19+
See [issue1](https://github.com/longcw/yolo2-pytorch/issues/1)
20+
and [issue23](https://github.com/longcw/yolo2-pytorch/issues/23)
2021
for more details about training.
2122

22-
BTW, I recommend to write your own dataloader using [torch.utils.data.Dataset](http://pytorch.org/docs/data.html)
23-
since `multiprocessing.Pool.imap` won't stop even there is no enough memory space.
23+
**NOTE 2:**
24+
I recommend to write your own dataloader using [torch.utils.data.Dataset](http://pytorch.org/docs/data.html)
25+
since `multiprocessing.Pool.imap` won't stop even there is no enough memory space.
26+
An example of `dataloader` for VOCDataset: [issue71](https://github.com/longcw/yolo2-pytorch/issues/71).
2427

2528

2629

27-
### Installation and demo
30+
## Installation and demo
2831
1. Clone this repository
2932
```bash
3033
git clone git@github.com:longcw/yolo2-pytorch.git
@@ -39,7 +42,7 @@ since `multiprocessing.Pool.imap` won't stop even there is no enough memory spac
3942
and set the model path in `demo.py`
4043
4. Run demo `python demo.py`.
4144

42-
### Training YOLOv2
45+
## Training YOLOv2
4346
You can train YOLO2 on any dataset. Here we train it on VOC2007/2012.
4447

4548
1. Download the training, validation, test data and VOCdevkit
@@ -90,15 +93,15 @@ and set the path in `yolo2-pytorch/cfgs/exps/darknet19_exp1.py`.
9093
6. Run the training program: `python train.py`.
9194

9295

93-
### Evaluation
96+
## Evaluation
9497

9598
Set the path of the `trained_model` in `yolo2-pytorch/cfgs/config.py`.
9699
```bash
97100
cd faster_rcnn_pytorch
98101
mkdir output
99102
python test.py
100103
```
101-
### Training on your own data
104+
## Training on your own data
102105

103106
The forward pass requires that you supply 4 arguments to the network:
104107

0 commit comments

Comments
 (0)