Update mdr_pdf_parser.py
Browse files- mdr_pdf_parser.py +2 -2
mdr_pdf_parser.py
CHANGED
@@ -31,7 +31,7 @@ import torch # PyTorch
|
|
31 |
import requests # For downloading models
|
32 |
from pathlib import Path
|
33 |
from enum import auto, Enum
|
34 |
-
from dataclasses import dataclass
|
35 |
from typing import Literal, Iterable, Generator, Sequence, Callable, TypeAlias, List, Dict, Any, Optional
|
36 |
from collections import defaultdict
|
37 |
from math import pi, ceil, sin, cos, sqrt, atan2
|
@@ -1376,7 +1376,7 @@ class _MDR_ONNXParams:
|
|
1376 |
cls_image_shape: str = "3,48,192"
|
1377 |
cls_batch_num: int = 6
|
1378 |
cls_thresh: float = 0.9
|
1379 |
-
label_list:
|
1380 |
|
1381 |
# Attributes with default values (Group 2 - Detection)
|
1382 |
det_algorithm: str = "DB"
|
|
|
31 |
import requests # For downloading models
|
32 |
from pathlib import Path
|
33 |
from enum import auto, Enum
|
34 |
+
from dataclasses import dataclass, field
|
35 |
from typing import Literal, Iterable, Generator, Sequence, Callable, TypeAlias, List, Dict, Any, Optional
|
36 |
from collections import defaultdict
|
37 |
from math import pi, ceil, sin, cos, sqrt, atan2
|
|
|
1376 |
cls_image_shape: str = "3,48,192"
|
1377 |
cls_batch_num: int = 6
|
1378 |
cls_thresh: float = 0.9
|
1379 |
+
label_list: List[str] = field(default_factory=lambda: ['0', '180'])
|
1380 |
|
1381 |
# Attributes with default values (Group 2 - Detection)
|
1382 |
det_algorithm: str = "DB"
|