deepvis.py 文件源码

def browse_layer(self, batch_size=25, cols=5):
        def plot(layer_id, batch_id):
            plt.figure(figsize=(14,20))
            all_files = sorted(os.listdir('{}/{}'.format(self.save_dir_, layer_id)))

            batch_id = int(batch_id)
            img_list, label_list = [],[]
            for f in all_files[batch_id*batch_size:(batch_id+1)*batch_size]:
                img_path = os.path.join(self.save_dir_, layer_id, f, 'img.jpg')
                img = plt.imread(img_path)
                img_list.append(img)
                label_list.append(f)
            plot_list(img_list, label_list, cols_nr=cols)
        return interact(plot, layer_id='17',batch_id='6')