aminehorseman / mobilenet-v2-custom-dataset Goto Github PK

Issue: Name of the directory in the data set is not clear. Should the names of the directory be a string or should it be number?

My directory structure for the data set (as mentioned in the README.md):

data\
    class1\
        img1.jpg
        img2.jpg
        img3.jpg
        ...
    class1\
        img1.jpg
        img2.jpg
        img3.jpg
        ...
    .
    .
    .

In the utils.py
Method: generate_batches
You use int(files[i].split('/')[1]), does that mean you want to cast the names to int? If so this is throwing an error.

**Error**: invalid literal for int() with base 10: 'class1'
Configurations:

• Tensorflow Version: 1.13.1
• Python: 3.7.3
• OS: Mac OS Catalina 10.15.3