Skip to content

Commit 0f6def2

Browse files
authored
Merge pull request longcw#36 from ml9951/update-readme
Adding instructions for training on other data sets
2 parents edda7d4 + e1b7eca commit 0f6def2

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

README.md

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,3 +97,14 @@ cd faster_rcnn_pytorch
9797
mkdir output
9898
python test.py
9999
```
100+
### Training on your own data
101+
102+
The forward pass requires that you supply 4 arguments to the network:
103+
104+
- `im_data` - image data.
105+
- This should be in the format `C x H x W`, where `C` corresponds to the color channels of the image and `H` and `W` are the height and width respectively.
106+
- Color channels should be in RGB format.
107+
- Use the `imcv2_recolor` function provided in `utils/im_transform.py` to preprocess your image. Also, make sure that images have been resized to `416 x 416` pixels
108+
- `gt_boxes` - A list of `numpy` arrays, where each one is of size `N x 4`, where `N` is the number of features in the image. The four values in each row should correspond to `x_bottom_left`, `y_bottom_left`, `x_top_right`, and `y_top_right`.
109+
- `gt_classes` - A list of `numpy` arrays, where each array contains an integer value corresponding to the class of each bounding box provided in `gt_boxes`
110+
- `dontcare` - a list of lists

0 commit comments

Comments
 (0)