Update README.md
Browse files
README.md
CHANGED
|
@@ -20,7 +20,7 @@ tags:
|
|
| 20 |
## Author
|
| 21 |
- **Vishal Adithya.A**
|
| 22 |
## Overview
|
| 23 |
-
This project demonstrates a depth estimation XgBoost Regressor model that predicts the average depth of images provided using features extracted from a pre-trained ResNet50 model.The model was trained upon the **NYUv2 dataset** ([0jl/NYUv2](https://huggingface.co/datasets/0jl/NYUv2)). The trained model is saved
|
| 24 |
|
| 25 |
### Loading the Model
|
| 26 |
The model is saved as `model.pkl` using `pickle`. You can load and use it as follows:
|
|
|
|
| 20 |
## Author
|
| 21 |
- **Vishal Adithya.A**
|
| 22 |
## Overview
|
| 23 |
+
This project demonstrates a depth estimation XgBoost Regressor model that predicts the average depth of images provided using features extracted from a pre-trained ResNet50 model.The model was trained upon the **NYUv2 dataset** ([0jl/NYUv2](https://huggingface.co/datasets/0jl/NYUv2)). The trained model is saved using Python's `pickle` library for easy deployment and reuse.
|
| 24 |
|
| 25 |
### Loading the Model
|
| 26 |
The model is saved as `model.pkl` using `pickle`. You can load and use it as follows:
|