Я пытаюсь запустить FienSoP / canny_edge_detector код на github с моим собственным каталогом. Но вместо того, чтобы показывать фигуру, она показывает размер фигуры ().
canny_Edge_Detector.ipynb
%matplotlib inline
from utils import utils
import canny_edge_detector as ced
imgs = utils.load_data('/Users/office/Mask_RCNN/office/images')
utils.visualize(imgs, 'gray')
<Figure size 1440x2880 with 0 Axes>
util.py
import numpy as np
import skimage
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import os
import scipy.misc as sm
import glob
def rgb2gray(rgb):
r, g, b = rgb[:,:,0], rgb[:,:,1], rgb[:,:,2]
gray = 0.2989 * r + 0.5870 * g + 0.1140 * b
return gray
def load_data(dir_name):
imgs = []
data_path = os.path.join(dir_name,'*g')
files = glob.glob(data_path)
for filename in files:
img = mpimg.imread(filename)
img = rgb2gray(img)
imgs.append(img)
return imgs
def visualize(imgs, format=None, gray=False):
plt.figure(figsize=(20, 40))
for i, img in enumerate(imgs):
if img.shape[0] == 3:
img = img.transpose(1,2,0)
plt_idx = i+1
plt.subplot(len(imgs), 1, plt_idx)
plt.imshow(img, format)
plt.show()
Так как, я начинающий, описанные ответы будут очень полезны. Заранее спасибо.