BhumikaMak commited on
Commit
9d4edcb
·
verified ·
1 Parent(s): 404690c

Update yolov5.py

Browse files
Files changed (1) hide show
  1. yolov5.py +1 -1
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.5)
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