From f08eb0451203004e377c3e0a7fdd0f7d57aeff5f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?L=C3=AA=20Trung=20Hi=E1=BA=BFu?= <30469654+hieuck@users.noreply.github.com> Date: Wed, 4 Dec 2024 22:59:04 +0700 Subject: [PATCH] Update build_backbone.py --- models/backbones/build_backbone.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/backbones/build_backbone.py b/models/backbones/build_backbone.py index ed1e6ea..daed15a 100644 --- a/models/backbones/build_backbone.py +++ b/models/backbones/build_backbone.py @@ -5,7 +5,7 @@ from torchvision.models import vgg16, vgg16_bn, VGG16_Weights, VGG16_BN_Weights, from models.backbones.pvt_v2 import pvt_v2_b2, pvt_v2_b5 from models.backbones.swin_v1 import swin_v1_t, swin_v1_s, swin_v1_b, swin_v1_l from config import Config -from utils import check_download_model +from BiRefNetutils import check_download_model config = Config()