@@ -4,27 +4,30 @@ implementation of YOLOv2.
4
4
This project is mainly based on [ darkflow] ( https://github.com/thtrieu/darkflow )
5
5
and [ darknet] ( https://github.com/pjreddie/darknet ) .
6
6
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
-
11
7
I used a Cython extension for postprocessing and
12
8
` multiprocessing.Pool ` for image preprocessing.
13
9
Testing an image in VOC2007 costs about 13~ 20ms.
14
10
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:**
16
16
This is still an experimental project.
17
17
VOC07 test mAP is about 0.71 (trained on VOC07+12 trainval,
18
18
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 )
20
21
for more details about training.
21
22
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 ) .
24
27
25
28
26
29
27
- ### Installation and demo
30
+ ## Installation and demo
28
31
1 . Clone this repository
29
32
``` bash
30
33
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
39
42
and set the model path in ` demo.py`
40
43
4. Run demo ` python demo.py` .
41
44
42
- # ## Training YOLOv2
45
+ # # Training YOLOv2
43
46
You can train YOLO2 on any dataset. Here we train it on VOC2007/2012.
44
47
45
48
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`.
90
93
6. Run the training program: ` python train.py` .
91
94
92
95
93
- # ## Evaluation
96
+ # # Evaluation
94
97
95
98
Set the path of the ` trained_model` in ` yolo2-pytorch/cfgs/config.py` .
96
99
` ` ` bash
97
100
cd faster_rcnn_pytorch
98
101
mkdir output
99
102
python test.py
100
103
` ` `
101
- # ## Training on your own data
104
+ # # Training on your own data
102
105
103
106
The forward pass requires that you supply 4 arguments to the network:
104
107
0 commit comments