Skip to content

Commit 2cc22a2

Browse files
authored
Update README.md
1 parent 6febb3b commit 2cc22a2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@ Validation images = 115
1111

1212
## Preprocessing :
1313

14-
The Preprocess.py file handles all the necessary preprocessing and saves the data in the form of numpy arrays. First, the images are resized to (512,512) dimensions. Accordingly, the ground truth of the boxes is modified as well. All the images are normalized to a range of [-1 , 1]. The ground truth coordinates are processed to form a matrix of dimensions as ( grid height , grid width , 1 , 5 ).
14+
The `Preprocess.py` file handles all the necessary preprocessing and saves the data in the form of numpy arrays. First, the images are resized to (512,512) dimensions. Accordingly, the ground truth of the boxes is modified as well. All the images are normalized to a range of [-1 , 1]. The ground truth coordinates are processed to form a matrix of dimensions as ( grid height , grid width , 1 , 5 ).
1515

1616
### Custom data :
17-
Necessary changes need to be done in the Preprocess.py file to input the custom data and images, to create the appropriate input and output matrices.
17+
Necessary changes need to be done in the `Preprocess.py` file to input the custom data and images, to create the appropriate input and output matrices.
1818

1919
## Model :
2020

@@ -30,7 +30,7 @@ The model is trained for 180 epochs in total with a batch size of 4. The learni
3030

3131
## Inference :
3232

33-
The Utils.py consists of the functions used to convert the matrix output ( grid height , grid width , 1 , 5 ) of the model to actual predicted bounding boxes. Non max suppression is used to eliminate boxes on the same object as stated in the YOLO paper.
33+
The `Utils.py` consists of the functions used to convert the matrix output ( grid height , grid width , 1 , 5 ) of the model to actual predicted bounding boxes. Non max suppression is used to eliminate boxes on the same object as stated in the YOLO paper.
3434

3535
## Results :
3636

0 commit comments

Comments
 (0)