Skip to content

A unified codebase, developed in TensorFlow 2, for a collection of deep neural networks for matching, segmenting and translating acoustic and satellite images.

License

Notifications You must be signed in to change notification settings

giovgiac/neptune

Repository files navigation

Neptune

A unified codebase, programmed in Python 3, for executing a collection of neural networks being developed at Federal University of Rio Grande by one Computer Engineering student and a Ph.D student:

  • Mr. Giovanni Gatti De Giacomo
  • MSc Matheus Machado dos Santos

Networks

In the codebase there are three supported applications that run on neural networks:

  • Semantic segmentation of water and stationary and moving objects in satellite images (U-Net);
  • Matching between acoustic and satellite images (DizygoticNet);
  • General translation of a large satellite image and acoustic image pair to a single equivalent satellite image (W-Net).

Dependencies

Currently, our codebase depends on the following packages:

  • abseil >= 0.7.1
  • namegenerator >= 1.0.6
  • pandas >= 0.24.2
  • scikit-learn >= 0.21.3
  • tensorflow >= 2.0.0
  • tqdm >= 4.36.1

Execution

To choose the neural network and dataset combination to use, the file 'main.py' needs to be edited. If you need to change default configuration, such as shape or batch size, then you will also want to edit 'config.py'.

After setting up the appropriate networks, datasets and configurations, execution is as simple as:

$ python general.py

The command for tensorboard will be automatically generated and printed by the program in the terminal.

Configuration can also be specified via command-line arguments. For a list of all possibilities take a look at 'config.py', an example is provided below:

$ python general.py --batch_size=16 --learning_rate=1e-4

Publications

Below is a list of papers we published that are related to this codebase:

Datasets

About

A unified codebase, developed in TensorFlow 2, for a collection of deep neural networks for matching, segmenting and translating acoustic and satellite images.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages