ml4a / ml4a

A python library and collection of notebooks for making art with machine learning.

Home Page:https://ml4a.net

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Issue with stylegan.dataset_tool transform parameter

Solcany opened this issue · comments

I'm following this ml4a notebook

When I try to run this block:

from ml4a.models import stylegan
  
  # replace 'images_folder' and 'dataset_output' with your own locations.
  
  config = {
      'images_folder': '/pathToMyDataset/',
      'dataset_output': '/pathToDatasetDest/',
      'transform': None,
      'labels': False,
      'size': 256
  }
  
  stylegan.dataset_tool(config)

I get:

TypeError: sequence item 7: expected str instance, NoneType found

This error only occurs when I use 'transform': None.

This should be fixed with the next update to the pip library. for now, you can fix this by installing ml4a from source, i.e.

instead of pip install ml4a, do pip install git+https://github.com/ml4a/ml4a.