dt-geo/PandSWavesDetectionTool
2
0
Fork 0
forked from epos-ai/epos-ai-picking-tools
Code Pull Requests Activity

Add new weights

Browse Source
This commit is contained in:
Hubert Siejkowski
2024-07-19 16:11:36 +02:00
parent e6b20a7567
commit 74cdf27dd6
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/epos_ai_picking_tools/phasenet.py
View File

@@ -13,7 +13,7 @@ EPOS_AI_MODEL_REPOSIOTRY_URL = "https://models.isl.grid.cyfronet.pl/models/v3/"
class PhaseNetModelRunner(ModelRunner):
model_type = "PhaseNet"
extra_weights = [("bogdanka", "1"), ("lgcd", "1"), ("bogdanka_lgcd", "1")]
extra_weights = [("bogdanka", "1"), ("lgcd", "1"), ("bogdanka_lgcd", "1"), ("sentinels", "1")]
def __init__(self, weights_name="original", output_dir=pathlib.Path("."), **kwargs):
self.model_name = getattr(sbm, PhaseNetModelRunner.model_type)