vittoriopippi
commited on
Commit
·
ab7810d
1
Parent(s):
e981486
Fix import errors
Browse files- modeling_vatrpp.py +3 -13
modeling_vatrpp.py
CHANGED
@@ -1,24 +1,14 @@
|
|
1 |
from transformers import PreTrainedModel, AutoModel, AutoConfig
|
2 |
from .configuration_vatrpp import VATrPPConfig
|
3 |
-
import json
|
4 |
import os
|
5 |
-
import random
|
6 |
-
import shutil
|
7 |
-
from collections import defaultdict
|
8 |
-
import time
|
9 |
-
from datetime import timedelta
|
10 |
-
from pathlib import Path
|
11 |
|
12 |
import cv2
|
13 |
import numpy as np
|
14 |
import torch
|
15 |
|
16 |
-
from data.dataset import FolderDataset
|
17 |
-
from models.model import VATr
|
18 |
-
from util.
|
19 |
-
from util.misc import FakeArgs
|
20 |
-
from util.text import TextGenerator
|
21 |
-
from util.vision import detect_text_bounds
|
22 |
from torchvision.transforms.functional import to_pil_image
|
23 |
|
24 |
|
|
|
1 |
from transformers import PreTrainedModel, AutoModel, AutoConfig
|
2 |
from .configuration_vatrpp import VATrPPConfig
|
|
|
3 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
|
5 |
import cv2
|
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 |
|