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"
|