Update oct_restore.py

sishenyuan · Jun 27, 2024 · 901d191 · 901d191
1 parent cfeb504
commit 901d191
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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