Internal chnage

official/vision/beta/projects/basnet/configs/experiments/basnet_dut_gpu.yaml

@@ -3,7 +3,7 @@ runtime:
   mixed_precision_dtype: 'float32'
   num_gpus: 8
 task:
-  init_checkpoint: '/home/gunho1123/ckpt/ckpt_encoder_wbias01'
+  init_checkpoint: 'gs://cloud-basnet-checkpoints/basnet_encoder_imagenet/ckpt-340306'
   init_checkpoint_modules: 'backbone'
   train_data:
     dtype: 'float32'

official/vision/beta/projects/basnet/evaluation/metrics.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""
+"""Evaluation metrics for BASNet.
 
 The MAE and maxFscore implementations are a modified version of
 https://github.com/xuebinqin/Binary-Segmentation-Evaluation-Tool

official/vision/beta/projects/basnet/losses/basnet_losses.py

@@ -55,7 +55,6 @@ class BASNetLoss:
     total_loss = total_loss / len(levels)
 
     return total_loss
-    
 
   def _iou_loss(self, sigmoids, labels):
     total_iou_loss = 0
@@ -66,7 +65,3 @@ class BASNetLoss:
     total_iou_loss += 1-IoU
 
     return total_iou_loss
-
-
-
-

official/vision/beta/projects/basnet/modeling/refunet.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 
-# Import libraries
 import tensorflow as tf
 from official.vision.beta.projects.basnet.modeling.layers import nn_blocks