Skip to content

An Android app for DeRaining Images Using pix2pix GAN model architecture

License

Notifications You must be signed in to change notification settings

Abhi03-ag/Image-DeRaining-App

Repository files navigation

Image DeRaining Android App

This is an Android sample app with the pix2pix GAN TensorFlow Lite model for removing rain-streaks from rainy images.

NoteBook

Refer Notebook folder for pix2pix GAN model source code - paper.

Dataset used Link.

TensorFlow Lite Model

There is one TensorFlow Lite Models included Link (Refer README for details).
Android Studio ML Model Binding was used to import these models into the Android project.

Requirements

  • Android Studio Preview Beta version - download here.
  • Android device in developer mode with USB debugging enabled
  • USB cable to connect an Android device to computer

Build and run

  • Clone the project repo:
    git clone https://github.com/Abhi03-ag/Image-DeRaining-App.git
  • Open the Android code android/CaptureDeRainedImages in android Studio.
  • Download the TFLite model from the google drive link mentioned above.
  • Create a new asset folder named ml in app/src/main and add the tflite model there.
  • Connect your Android device to computer then click on "Run -> Run 'app'.
  • Once the app is launched on device, grant camera permission.
  • Use the App

App Screenshot

HomeScreen ResultScreen

Releases

No releases published

Packages

No packages published