From 901d1914204dd4c5c46d4cdd1677bd2a4a36d210 Mon Sep 17 00:00:00 2001
From: sishenyuan <78411486+sishenyuan@users.noreply.github.com>
Date: Thu, 27 Jun 2024 20:58:57 +0800
Subject: [PATCH] Update oct_restore.py

---
 src/oct_restore.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/oct_restore.py b/src/oct_restore.py
index c351840..9dfc625 100644
--- a/src/oct_restore.py
+++ b/src/oct_restore.py
@@ -12,8 +12,8 @@
 from PIL import Image, ImageOps
 from torchvision import transforms
 
-from yolo.detect import detect_nurd
-from net3inference import ImageVectorDataset, VisionTransformer
+from ndnet.detect import detect_nurd
+from ncnet import ImageVectorDataset, VisionTransformer
 from torch.utils.data import DataLoader
 import time
 import argparse