Skip to content
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

add createRandomDataset in Augmentation #21

Merged
merged 4 commits into from
Sep 30, 2020

Conversation

luisaheise
Copy link
Member

Criamos a opção de gerar um dataset no formato especificado apenas forncendo
a pasta que contem as pastas com pngs de pokemons e a pasta com paisagens.

Também inclui meu nome e o da Camilala no readme

Criamos a opção de gerar um dataset no formato especificado apenas forncendo
a pasta que contem as pastas com pngs de pokemons e a pasta com paisagens.

Também inclui meu nome e o da Camilala no readme

for i in range(numPics):
#escolher quantos pokemons tera nessa imagem
num_pokemons = np.random.randint(low=num_min_pokemons, high=num_max_pokemons)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

dá pra colocar um bias aqui? pra ter mais chance de ter o mínimo de pokémons e menos chance de ter o máximo

- the script now chooses compatible background and pokemon sizes, so that is
not causing erros anymore

- the ammount of pokemons in the images is now chosen based on a normal
distribution
@@ -431,3 +439,73 @@ def applyMask(img_path, n_its=5, step=1, img_show=False):
plt.show()
return img_masked

def generateRandomDatatset(numPics, pokemons_path, backgrounds_path, mean_number_pokemons=2, std_number_pokemons=1):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Só curiosidade, por que o nome generateRandomDatatset e não generateRandomDataSet?

Utils/Augmentation/__init__.py Outdated Show resolved Hide resolved
Utils/Augmentation/__init__.py Outdated Show resolved Hide resolved
Utils/Augmentation/__init__.py Show resolved Hide resolved
@luisaheise luisaheise merged commit a949396 into master Sep 30, 2020
@anor4k anor4k deleted the pokemon_augmentation_improvment branch February 12, 2021 21:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants