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
|