dt-geo/PandSWavesDetectionTool
1
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -22,6 +22,7 @@ classifiers = [
"Programming Language :: Python :: Implementation :: CPython",
]
dependencies = [
"numpy<2.0.0",
"seisbench==0.5.*",
"click"
]

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)