This src is used for Kaggle-Shopee Code League https://www.kaggle.com/competitions/shopee-product-detection-open

Pretrained model:

• Efficient noisy student

  • tf_efficientnet_b0_ns
  • tf_efficientnet_b1_ns
  • tf_efficientnet_b2_ns
  • tf_efficientnet_b3_ns
  • tf_efficientnet_b4_ns
  • tf_efficientnet_b5_ns
  • tf_efficientnet_b6_ns
  • tf_efficientnet_b7_ns

• Efficient net

  • efficientnet_b0
  • efficientnet_b1
  • efficientnet_b2
  • efficientnet_b3
  • efficientnet_b4
  • efficientnet_b5
  • efficientnet_b6
  • efficientnet_b7

Specify model: model = tf_efficientnet_b4_ns(True)

Specify data path:

• train_set = ProductImageLoader(image_folder, train_csv_file, 'train')
• val_set = ProductImageLoader(image_folder, val_csv_file, 'val')