def bar_plot(dir): x=len(os.listdir(dir+'/ok')) y=len(os.listdir(dir+'/nok')) category = ['ok', 'nok'] count = [x, y] plot = plt.bar(category,count) plot[0].set_color('green') plt.title('Plot of number of values for each category') plt.show() print('Training images:\n') print('OK:', len(os.listdir(train_dir+'/ok'))) print('NOK:', len(os.listdir(train_dir+'/nok'))) print('Total Training images:', len(os.listdir(train_dir+'/ok')) + len(os.listdir(train_dir+'/nok'))) print('*'*49) bar_plot(train_dir) print('Testing images:\n') print('OK:', len(os.listdir(test_dir+'/ok'))) print('NOK:', len(os.listdir(test_dir+'/nok'))) print('Total Testing images:', len(os.listdir(test_dir+'/ok')) + len(os.listdir(test_dir+'/nok'))) print('*'*49) bar_plot(test_dir) print('Validation images:\n') print('OK:', len(os.listdir(val_dir+'/ok'))) print('NOK:', len(os.listdir(val_dir+'/nok'))) print('Total Testing images:', len(os.listdir(val_dir+'/ok')) + len(os.listdir(val_dir+'/nok'))) print('*'*49) bar_plot(val_dir)
Preview:
downloadDownload PNG
downloadDownload JPEG
downloadDownload SVG
Tip: You can change the style, width & colours of the snippet with the inspect tool before clicking Download!
Click to optimize width for Twitter