Update mirnet.py (#1172)

ADITYADAS1999 · web-flow · commit 17a38bba9f27 · 2022-12-27T15:14:57.000+01:00
* Update mirnet.py

* Update mirnet.md

* Update mirnet.ipynb
diff --git a/examples/vision/ipynb/mirnet.ipynb b/examples/vision/ipynb/mirnet.ipynb
@@ -444,7 +444,7 @@
     "        up_sampling_module(dual_attention_unit_block(level3_skff))\n",
     "    )\n",
     "    # SKFF 2\n",
-    "    skff_ = selective_kernel_feature_fusion(level1_dau_2, level3_dau_2, level3_dau_2)\n",
+    "    skff_ = selective_kernel_feature_fusion(level1_dau_2, level2_dau_2, level3_dau_2)\n",
     "    conv = layers.Conv2D(channels, kernel_size=(3, 3), padding=\"same\")(skff_)\n",
     "    return layers.Add()([input_tensor, conv])\n",
     ""
@@ -669,4 +669,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
+}
diff --git a/examples/vision/md/mirnet.md b/examples/vision/md/mirnet.md
@@ -375,7 +375,7 @@ def multi_scale_residual_block(input_tensor, channels):
         up_sampling_module(dual_attention_unit_block(level3_skff))
     )
     # SKFF 2
-    skff_ = selective_kernel_feature_fusion(level1_dau_2, level3_dau_2, level3_dau_2)
+    skff_ = selective_kernel_feature_fusion(level1_dau_2, level2_dau_2, level3_dau_2)
     conv = layers.Conv2D(channels, kernel_size=(3, 3), padding="same")(skff_)
     return layers.Add()([input_tensor, conv])
 
@@ -687,4 +687,4 @@ for low_light_image in random.sample(test_low_light_images, 6):
 
 
 
-![png](/img/examples/vision/mirnet/mirnet_21_5.png)
+![png](/img/examples/vision/mirnet/mirnet_21_5.png)
diff --git a/examples/vision/mirnet.py b/examples/vision/mirnet.py
@@ -350,7 +350,7 @@ def multi_scale_residual_block(input_tensor, channels):
         up_sampling_module(dual_attention_unit_block(level3_skff))
     )
     # SKFF 2
-    skff_ = selective_kernel_feature_fusion(level1_dau_2, level3_dau_2, level3_dau_2)
+    skff_ = selective_kernel_feature_fusion(level1_dau_2, level2_dau_2, level3_dau_2)
     conv = layers.Conv2D(channels, kernel_size=(3, 3), padding="same")(skff_)
     return layers.Add()([input_tensor, conv])
 

Original file line number	Diff line number	Diff line change
`@@ -350,7 +350,7 @@ def multi_scale_residual_block(input_tensor, channels):`
`350`	`350`	`up_sampling_module(dual_attention_unit_block(level3_skff))`
`351`	`351`	`)`
`352`	`352`	`# SKFF 2`
`353`		`- skff_ = selective_kernel_feature_fusion(level1_dau_2, level3_dau_2, level3_dau_2)`
	`353`	`+ skff_ = selective_kernel_feature_fusion(level1_dau_2, level2_dau_2, level3_dau_2)`
`354`	`354`	`conv = layers.Conv2D(channels, kernel_size=(3, 3), padding="same")(skff_)`
`355`	`355`	`return layers.Add()([input_tensor, conv])`
`356`	`356`