vishal-adithya commited on
Commit
d205d57
·
verified ·
1 Parent(s): d7bb60d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -10
README.md CHANGED
@@ -13,29 +13,29 @@ pipeline_tag: image-segmentation
13
 
14
  ## Overview
15
 
16
- This model demonstrates a road segmentation implemented using deep learning techniques which predics the road regions in the input image and returns it in a grayscale image.
17
 
18
  ## Model Structure
19
- - **rs1-low.pth**: The lowest performer model with a loss of 0.69%.
20
- - **rs1-high.pth**: The highest performer model with a loss of 0.07%.
21
 
22
  ## Features
23
  1. ### Architecture
24
- - Basic Resnet50 model with few upsampling and batch normalisation layers.
25
- - Contains over _____ Trainable paramameters.
26
- - Training Duration: 1 hour.
27
  ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6787e9bb4151553bf9307186/rdnXew3tWUVGoKXhRK8SX.png)
28
  2. ### Training Data
29
  - Source: ([bnsapa/road-detection](https://huggingface.co/datasets/bnsapa/road-detection))
30
- - Format: The dataset includes RGB images of roads around the globe and thrie corresponding segment and lane.
31
  - Preprocessing: With the help of torch and torchvission api basic preprocessing like resizing and convertion to tensor were implemented.
32
  3. ### CostFunctions Score
33
  - BCE:
34
  - MSE:
35
 
36
  ## License
37
- This project is licensed under the Apache License 2.0.
38
 
39
  ## Acknowledgments
40
- - Apple M1 Pro 16gb of unfied memory for efficient GPU acceleration during model training
41
- - Pytorch for robust deep learning framework
 
13
 
14
  ## Overview
15
 
16
+ This model demonstrates a road segmentation implemented using **deep learning** techniques which predicts the road regions in the input image and returns it in a grayscale format.
17
 
18
  ## Model Structure
19
+ - **rs1-low.pth**: The lowest performer model with a loss of **0.69%**.
20
+ - **rs1-high.pth**: The highest performer model with a loss of **0.07%**.
21
 
22
  ## Features
23
  1. ### Architecture
24
+ - Basic **Resnet50** model with few upsampling and batch normalisation layers.
25
+ - Contains over ** ** Trainable paramameters.
26
+ - Training Duration: **1 hour**.
27
  ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6787e9bb4151553bf9307186/rdnXew3tWUVGoKXhRK8SX.png)
28
  2. ### Training Data
29
  - Source: ([bnsapa/road-detection](https://huggingface.co/datasets/bnsapa/road-detection))
30
+ - Format: The dataset includes RGB images of roads around the globe and their corresponding segment and lane.
31
  - Preprocessing: With the help of torch and torchvission api basic preprocessing like resizing and convertion to tensor were implemented.
32
  3. ### CostFunctions Score
33
  - BCE:
34
  - MSE:
35
 
36
  ## License
37
+ This project is licensed under the **Apache License 2.0**.
38
 
39
  ## Acknowledgments
40
+ - **Apple M1 Pro 16gb** of unfied memory for efficient GPU acceleration during model training
41
+ - **Pytorch** for robust deep learning framework