vittoriopippi commited on
Commit ·
7248816
1
Parent(s): 07f8685
Change imports
Browse files- modeling_vatrpp.py +3 -3
modeling_vatrpp.py
CHANGED
|
@@ -6,9 +6,9 @@ import cv2
|
|
| 6 |
import numpy as np
|
| 7 |
import torch
|
| 8 |
|
| 9 |
-
from
|
| 10 |
-
from
|
| 11 |
-
from
|
| 12 |
from torchvision.transforms.functional import to_pil_image
|
| 13 |
|
| 14 |
|
|
|
|
| 6 |
import numpy as np
|
| 7 |
import torch
|
| 8 |
|
| 9 |
+
from data.dataset import FolderDataset
|
| 10 |
+
from models.model import VATr
|
| 11 |
+
from util.vision import detect_text_bounds
|
| 12 |
from torchvision.transforms.functional import to_pil_image
|
| 13 |
|
| 14 |
|