-
Notifications
You must be signed in to change notification settings - Fork 402
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Rgb noise models #357
Rgb noise models #357
Conversation
…itat-sim into redwood-noisy-depth
…itat-sim into redwood-noisy-depth
… redwood-noisy-depth
… redwood-noisy-depth
@@ -0,0 +1,72 @@ | |||
import abc |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I believe this file can be removed
viewer.py
Outdated
@@ -0,0 +1,87 @@ | |||
import cv2 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please remove this file from the PR
sensor_test.py
Outdated
@@ -0,0 +1,61 @@ | |||
import timeit |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Same with this file
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you add a test like https://github.com/facebookresearch/habitat-sim/blob/master/tests/test_sensors.py#L150 for the new noise models?
Deleting from PR.
Deleting from PR.
Removing from PR.
Merging origin/master to PR
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Added noise models RGB outputs with default parameters to make noise mode choice easier. |
Fix video generation
Addition of Gaussian noise, Speckle noise, Salt and Pepper noise and Poisson noise for RGB sensors.
Motivation and Context
This change builds upon the redwood depth noise model implemented by Erik Wijmans with the objective of implementing color noise models in Habitat. This change includes initial commits for four new RGB noise models, namely Gaussian noise, Speckle noise, Salt and Pepper noise and Poisson noise. The hope for this pull request is to continue the efforts to allow for embodied agents to train under realistic sensor noise to further understand their behavior in sub-optimal sensor conditions.
Noise models RGB output with default parameters (added by @mathfac)
![image](https://user-images.githubusercontent.com/877440/74765569-b7971700-5238-11ea-9971-6679cdca8c7d.png)
Gaussian Noise Model:
Poisson Noise Model:
![image](https://user-images.githubusercontent.com/877440/74766164-b1ee0100-5239-11ea-84cc-704cbf7b3258.png)
Speckle Noise Model:
![image](https://user-images.githubusercontent.com/877440/74766221-ca5e1b80-5239-11ea-99bd-863444af51ee.png)
SaltAndPepper Noise Model:
![image](https://user-images.githubusercontent.com/877440/74765590-c2ea4280-5238-11ea-8ea9-8c700d93ae1b.png)
How Has This Been Tested
These modification have been tested to gauge the frame rate at which the noise models are applied to arbitrary NumPy arrays as well as tested to see the visual effect of the noise models in the simulation environment. These tests can be replicated using sensor_test.py and viewer.py respectively.
Types of changes
Checklist