TALIE-1535: add orientation classification#19
Conversation
NastyBoget
force-pushed
the
TALIE-1535_orientation_classifier
branch
4 times, most recently
from
c37cb0c to
468175d
Compare
NastyBoget
force-pushed
the
TALIE-1535_orientation_classifier
branch
2 times, most recently
from
ff8839e to
4334237
Compare
NastyBoget
force-pushed
the
TALIE-1535_orientation_classifier
branch
from
4334237 to
877ae0e
Compare
| """ | ||
| super(ClassificationModelTorch, self).__init__() | ||
| self.efficientnet_b0 = models.efficientnet_b0(pretrained=model_path is None) | ||
| self.efficientnet_b0.classifier[1] = nn.Linear(in_features=1280, out_features=num_classes) |
There was a problem hiding this comment.
а тут в классификатор веса же не загружаются?
There was a problem hiding this comment.
Не загружаются, они загружаются в OrientationClassifier.
Если model_path не передавать, то будет загружаться pretrained-модель, это используется при обучении новой модели
| """ | ||
| def __init__(self, model_path: Optional[str], num_classes: int = 6) -> None: | ||
| """ | ||
| first 2 classes mean columns number |
There was a problem hiding this comment.
если я правильно понял, тут mean используется в качестве означает? мб написать the first 2 classes are the number of columns или the first 2 classes contain the number of columns
There was a problem hiding this comment.
и для других все равно не оч множет быть понятно, поч распределение классов именно такое, мб добавить пояснялку типа [1 колонка, несколько колонок, 0, 90, 180, 270]
There was a problem hiding this comment.
написала более подробный комментарий, и правда было не понятно
2 participants
No description provided.