Update README.md
Browse files
README.md
CHANGED
|
@@ -33,7 +33,7 @@ features = extract_features("path/to/image.jpg")
|
|
| 33 |
predicted_depth = model.predict([features])
|
| 34 |
print(predicted_depth[0])
|
| 35 |
```
|
| 36 |
-
**NOTE:** extract_features() is a predefined function in the original code which uses ResNet50 to extract features out of the image
|
| 37 |
|
| 38 |
## Key Features
|
| 39 |
- **Model Architecture**:
|
|
@@ -105,7 +105,7 @@ If you want to retrain the model, follow these steps:
|
|
| 105 |
with open("model.pkl", "wb") as f:
|
| 106 |
pickle.dump(regressor, f)
|
| 107 |
```
|
| 108 |
-
**NOTE:** This pipeline has just the base fundamental code more additional parameter tunings and preprocessing steps were being conducted during the training of the original model
|
| 109 |
|
| 110 |
|
| 111 |
## License
|
|
|
|
| 33 |
predicted_depth = model.predict([features])
|
| 34 |
print(predicted_depth[0])
|
| 35 |
```
|
| 36 |
+
**NOTE:** extract_features() is a predefined function in the original code which uses ResNet50 to extract features out of the image.
|
| 37 |
|
| 38 |
## Key Features
|
| 39 |
- **Model Architecture**:
|
|
|
|
| 105 |
with open("model.pkl", "wb") as f:
|
| 106 |
pickle.dump(regressor, f)
|
| 107 |
```
|
| 108 |
+
**NOTE:** This pipeline has just the base fundamental code more additional parameter tunings and preprocessing steps were being conducted during the training of the original model.
|
| 109 |
|
| 110 |
|
| 111 |
## License
|