I can't skip 'for i in dataset' of the ‘get_the_length_of_dataset’ function in 'prepare.py'

calmiLovesAI / Basic_CNNs_TensorFlow2

A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).

MIT License

524 stars 178 forks source link

I can't skip 'for i in dataset' of the ‘get_the_length_of_dataset’ function in 'prepare.py' #2

Closed 2605759123 closed 4 years ago

2605759123 commented 4 years ago

I follow the instructions in READMD.md.But ‘ def get_the_length_of_dataset(dataset): count = 0 for i in dataset: count += 1 return count ’ looks like a dead cycle.

My Python version is 3.7 and tensorflow version is 2.0.0. Windows10

Have you ever met this situation and how can I solve it Thanks.

2605759123 commented 4 years ago

My dataset includes images in JPG, JPEG and PNG formats. Can this be supported?

2605759123 commented 4 years ago

Ah, there seems to be something wrong with my environment. Now there is no such mistake.