Update README.md

Author: geekysethi

README.md

@@ -28,25 +28,32 @@ pip install git+https://github.com/geekysethi/headpose_estimation
 
 You can also install with the `setup.py`
 
-## Simple API with Face Detection
+##  With Face Detection
 To perform detection you can simple use the following lines:
 
 ```python
-
-
 import cv2
 from headpose_estimation import Headpose
+headpose = Headpose()
+img = cv2.imread("path_to_im.jpg")
+detections,image = headpose.run(img)
+```
 
-if __name__ == "__main__":
+This will return a list of dictionary which looks like this `[{'bbox': [xmin, ymin, xmax, ymax], 'yaw': yaw_value, 'pitch': pitch_value, 'roll': roll_value}`
 
-	headpose = Headpose()
 
-    img = cv2.imread("path_to_im.jpg")
-	detections,image = headpose.run(img)
-```
+##  Without Face Detection
+To perform detection you can simple use the following lines:
 
-This will return a list of dictionary which looks like this `[{'bbox': [xmin, ymin, xmax, ymax], 'yaw': yaw_value, 'pitch': pitch_value, 'roll': roll_value}`
+```python
+import cv2
+from headpose_estimation import Headpose
+headpose = Headpose(face_detection=False)
+imgcrop = cv2.imread("path_to_im.jpg")
+detections,image = headpose.run(imgcrop)
+```
 
+In this case it will return a list of dictionary which looks like this `[{'yaw': yaw_value, 'pitch': pitch_value, 'roll': roll_value}`
 
 ## Dependncies
 * EfficientNet https://github.com/qubvel/efficientnet

headpose_estimation/headpose.py

@@ -107,7 +107,7 @@ def run(self,image):
 
 			yaw,pitch,roll = self.detect_headpose(image)
 
-			final_output.append({"bbox":np.array(current_bbox),"yaw":yaw,"pitch": pitch,"roll": roll})
+			final_output.append({"yaw":yaw,"pitch": pitch,"roll": roll})
 			if(self.draw):
 				height,width,_ = image.shape
 				x_min, y_min, x_max,y_max = 0,0,width,height