Skip to content

Need some modifications on data_augmentation.py #4

@wenmengliu

Description

@wenmengliu

Hi Eugene,

Currently, I am using data_tools on the LiTS2017 dataset. I notice that in the data_augmentation.py line 154-156 in the 'image_random_transform' function might need some modifications. Because, in the repo, the input_image shape is based on 'tensorflow' (n_w,n_h,n_c) but if the input_image shape is on 'theano'. Using this function directly will cause the error? I mean I set the input_image shape is on 'theano' in (n_c,n_w,n_h). So I think it might need modifications under this case (Actually I changed it when I used, I just wanna ask it whether I did it right or not). Please kindly correct me if I were wrong.

screen shot 2018-10-09 at 10 59 22 am

Cheers,
Crystal

Metadata

Metadata

Assignees

No one assigned

    Labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions