Spaces:
Sleeping
Sleeping
Update yolov5.py
Browse files
yolov5.py
CHANGED
@@ -242,7 +242,7 @@ def dff_nmf(image, target_lyr, n_components):
|
|
242 |
|
243 |
#temp = image_array.reshape((rgb_img_float.shape[0],rgb_img_float.shape[1]) )
|
244 |
#visualization = show_factorization_on_image(rgb_img_float, image_array.resize((rgb_img_float.shape)) , image_weight=0.3)
|
245 |
-
visualization = show_factorization_on_image(rgb_img_float, np.transpose(rgb_channels, (2, 0, 1)), image_weight=0.
|
246 |
results.append(visualization)
|
247 |
plt.clf()
|
248 |
#return image_array
|
|
|
242 |
|
243 |
#temp = image_array.reshape((rgb_img_float.shape[0],rgb_img_float.shape[1]) )
|
244 |
#visualization = show_factorization_on_image(rgb_img_float, image_array.resize((rgb_img_float.shape)) , image_weight=0.3)
|
245 |
+
visualization = show_factorization_on_image(rgb_img_float, np.transpose(rgb_channels, (2, 0, 1)), image_weight=0.7)
|
246 |
results.append(visualization)
|
247 |
plt.clf()
|
248 |
#return image_array
|