Skip to content

Commit

Permalink
add header to all files
Browse files Browse the repository at this point in the history
  • Loading branch information
AmirMardan committed Jul 20, 2023
1 parent 9dabab1 commit 444bb69
Show file tree
Hide file tree
Showing 24 changed files with 220 additions and 4 deletions.
2 changes: 1 addition & 1 deletion LICENSE
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
MIT License

Copyright (c) 2023 Amir Mardan
Copyright (c) 2023 Les solutions Géostack, Inc.

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import os
import sys
package_path = os.path.abspath(os.path.join(__file__, "../"))
Expand Down
9 changes: 9 additions & 0 deletions first_break_picking/data/__init__.py
Original file line number Diff line number Diff line change
@@ -1 +1,10 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================
from first_break_picking.data.data_segy2npy import save_shots_fb
10 changes: 10 additions & 0 deletions first_break_picking/data/_data_dc_del.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

from pathlib import Path
import os
import numpy as np
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/data/_data_fb.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import os
from typing import List, Tuple
import pandas as pd
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/data/_datasets.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

from torch.utils.data import Dataset, random_split, DataLoader, ConcatDataset
import os
from typing import Tuple, List, Union
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/data/data_generator.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import numpy as np
from pathlib import Path
import argparse
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/data/data_segy2npy.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import numpy as np
import os
from typing import List, Union, Tuple, Optional
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/data/data_utils.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import random
from math import ceil
from typing import Tuple, List, Any, Union, Optional
Expand Down
9 changes: 9 additions & 0 deletions first_break_picking/data/dataset.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,12 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================
from torch.utils.data import Dataset, random_split, DataLoader, ConcatDataset
import os
from typing import Tuple, List, Union
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/data/seis_model.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import os.path
from typing import Optional, Union, List, Tuple

Expand Down
9 changes: 9 additions & 0 deletions first_break_picking/data/signals/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================
10 changes: 10 additions & 0 deletions first_break_picking/default_values.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import torch
import os
from argparse import ArgumentParser, Namespace
Expand Down
9 changes: 9 additions & 0 deletions first_break_picking/saved_checkpoints/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================
10 changes: 10 additions & 0 deletions first_break_picking/tools.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
Expand Down
9 changes: 9 additions & 0 deletions first_break_picking/train_eval/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================
10 changes: 10 additions & 0 deletions first_break_picking/train_eval/_predicting_tools.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

from typing import Optional, Union, Tuple, List
import matplotlib.pyplot as plt
import pandas as pd
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/train_eval/ai_tools.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import numpy as np
import torch
from typing import Optional, Union, Tuple, List
Expand Down
12 changes: 10 additions & 2 deletions first_break_picking/train_eval/metrics.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import torch
import torch.nn as nn
import torch.nn.functional as F
Expand All @@ -9,8 +19,6 @@
from first_break_picking.train_eval.unet import UNet
import first_break_picking.train_eval.ai_tools as tools



class BDiceLoss(nn.Module):
def __init__(self, weight=None, size_average=True):
super(BDiceLoss, self).__init__()
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/train_eval/parameter_tools.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import os
import logging
import torch
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/train_eval/predict.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import torch
from pathlib import Path
import shutil
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/train_eval/train.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import torch
from first_break_picking.data.dataset import get_loaders
from pathlib import Path
Expand Down
10 changes: 10 additions & 0 deletions first_break_picking/train_eval/unet.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# -*- coding: utf-8 -*-
# =============================================================================
# Copyright (C) Les solutions géostack, Inc.
#
# This file is part of the First-Break Picking package.
# Licensed under the terms of the MIT License.
#
# https://github.com/geo-stack/first_break_picking
# =============================================================================

import torch.nn as nn
import torch
import torch.nn.functional as F
Expand Down
5 changes: 4 additions & 1 deletion readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -174,8 +174,11 @@ predictor.predict(
<div class="alert alert-block alert-warning">
<b>Warning:</b> If you define the sampling rate incorrrectly, you can't see its effects on the plot (Y-axis is time step), but the saved time in first-break folder will be wrong.
</div>
<br>


<!-- ## Issues and Questions -->
**Ackowledgment:**<br>
This work, developped by [Amir Mardan](https://github.com/AmirMardan), was supported by Mitacs through the Mitacs Elevate Program.


[Top](#top)
Expand Down

0 comments on commit 444bb69

Please sign in to comment.