vishal-adithya commited on
Commit
ded5c80
·
verified ·
1 Parent(s): 9f25259

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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