Update app.py
Browse files
app.py
CHANGED
|
@@ -157,8 +157,8 @@ def remove_bg(fl, s, l, v, h, l_t):
|
|
| 157 |
frame_c = (frame.astype(np.int16)-bg_diff).astype(np.uint8)
|
| 158 |
|
| 159 |
|
| 160 |
-
|
| 161 |
-
hue_s = np.zeros_like(
|
| 162 |
edges = cv2.Laplacian(cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY), cv2.CV_64F)
|
| 163 |
blur_s = np.zeros_like(edges)
|
| 164 |
for i in range(2, frame.shape[0]-2):
|
|
@@ -166,7 +166,7 @@ def remove_bg(fl, s, l, v, h, l_t):
|
|
| 166 |
d = edges[i-2:i+2, j-2:j+2].var()
|
| 167 |
blur_s[i,j] = d.astype(np.uint8)
|
| 168 |
|
| 169 |
-
d =
|
| 170 |
hue_s[i,j] = (d*16).astype(np.uint8)
|
| 171 |
|
| 172 |
print(fl)
|
|
@@ -177,7 +177,6 @@ def remove_bg(fl, s, l, v, h, l_t):
|
|
| 177 |
print(np.average(blur_s))
|
| 178 |
print(np.median(blur_s))
|
| 179 |
#remove regions of low saturation and lightness (get scene without shadow)
|
| 180 |
-
hsv_ = cv2.cvtColor(frame_c, cv2.COLOR_BGR2HSV)
|
| 181 |
print("saturation")
|
| 182 |
print(np.average(hsv_[:,:,1]))
|
| 183 |
print(np.median(hsv_[:,:,1]))
|
|
|
|
| 157 |
frame_c = (frame.astype(np.int16)-bg_diff).astype(np.uint8)
|
| 158 |
|
| 159 |
|
| 160 |
+
hsv_ = cv2.cvtColor(frame_c, cv2.COLOR_BGR2HSV)
|
| 161 |
+
hue_s = np.zeros_like(hsv_)
|
| 162 |
edges = cv2.Laplacian(cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY), cv2.CV_64F)
|
| 163 |
blur_s = np.zeros_like(edges)
|
| 164 |
for i in range(2, frame.shape[0]-2):
|
|
|
|
| 166 |
d = edges[i-2:i+2, j-2:j+2].var()
|
| 167 |
blur_s[i,j] = d.astype(np.uint8)
|
| 168 |
|
| 169 |
+
d = hsv_[i-1:i+1, j-1:j+1, 0].var()
|
| 170 |
hue_s[i,j] = (d*16).astype(np.uint8)
|
| 171 |
|
| 172 |
print(fl)
|
|
|
|
| 177 |
print(np.average(blur_s))
|
| 178 |
print(np.median(blur_s))
|
| 179 |
#remove regions of low saturation and lightness (get scene without shadow)
|
|
|
|
| 180 |
print("saturation")
|
| 181 |
print(np.average(hsv_[:,:,1]))
|
| 182 |
print(np.median(hsv_[:,:,1]))
|