dasnet 0.1.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (20) hide show
  1. dasnet-0.1.1/PKG-INFO +20 -0
  2. dasnet-0.1.1/dasnet/__init__.py +14 -0
  3. dasnet-0.1.1/dasnet/data/__init__.py +0 -0
  4. dasnet-0.1.1/dasnet/data/das.py +723 -0
  5. dasnet-0.1.1/dasnet/inference.py +154 -0
  6. dasnet-0.1.1/dasnet/model/__init__.py +0 -0
  7. dasnet-0.1.1/dasnet/model/backbone_utils.py +241 -0
  8. dasnet-0.1.1/dasnet/model/dasnet.py +86 -0
  9. dasnet-0.1.1/dasnet/model/fastrcnn.py +844 -0
  10. dasnet-0.1.1/dasnet/model/maskrcnn.py +512 -0
  11. dasnet-0.1.1/dasnet/model/roiheads.py +1340 -0
  12. dasnet-0.1.1/dasnet/predict.py +731 -0
  13. dasnet-0.1.1/dasnet/utils.py +488 -0
  14. dasnet-0.1.1/dasnet.egg-info/PKG-INFO +20 -0
  15. dasnet-0.1.1/dasnet.egg-info/SOURCES.txt +18 -0
  16. dasnet-0.1.1/dasnet.egg-info/dependency_links.txt +1 -0
  17. dasnet-0.1.1/dasnet.egg-info/requires.txt +14 -0
  18. dasnet-0.1.1/dasnet.egg-info/top_level.txt +1 -0
  19. dasnet-0.1.1/pyproject.toml +32 -0
  20. dasnet-0.1.1/setup.cfg +4 -0
dasnet-0.1.1/PKG-INFO ADDED
@@ -0,0 +1,20 @@
1
+ Metadata-Version: 2.4
2
+ Name: dasnet
3
+ Version: 0.1.1
4
+ Summary: DASNet: Mask R-CNN for event detection in Distributed Acoustic Sensing data
5
+ License: MIT
6
+ Requires-Python: >=3.8
7
+ Description-Content-Type: text/markdown
8
+ Requires-Dist: torch
9
+ Requires-Dist: torchvision
10
+ Requires-Dist: numpy
11
+ Requires-Dist: scipy
12
+ Requires-Dist: pandas
13
+ Requires-Dist: h5py
14
+ Requires-Dist: tqdm
15
+ Requires-Dist: fsspec
16
+ Provides-Extra: train
17
+ Requires-Dist: obspy; extra == "train"
18
+ Requires-Dist: wandb; extra == "train"
19
+ Requires-Dist: gcsfs; extra == "train"
20
+ Requires-Dist: pycocotools; extra == "train"
dasnet-0.1.1/dasnet/__init__.py ADDED
@@ -0,0 +1,14 @@
1
+ from dasnet.inference import (
2
+ build_dasnet_model,
3
+ load_checkpoint,
4
+ make_infer_dataloader,
5
+ forward_raw,
6
+ postprocess_batch,
7
+ filter_by_score,
8
+ extract_peaks_for_instances,
9
+ default_device,
10
+ label_map,
11
+ save_predictions_json,
12
+ plot_das_predictions,
13
+ )
14
+ from dasnet.data.das import preprocess_data_rgb, preprocess_from_array
dasnet-0.1.1/dasnet/data/__init__.py ADDED
File without changes