(img_path)
| 53 | model = GCN(placeholders, logging=True) |
| 54 | |
| 55 | def load_image(img_path): |
| 56 | img = io.imread(img_path) |
| 57 | if img.shape[2] == 4: |
| 58 | img[np.where(img[:,:,3]==0)] = 255 |
| 59 | img = transform.resize(img, (224,224)) |
| 60 | img = img[:,:,:3].astype('float32') |
| 61 | |
| 62 | return img |
| 63 | |
| 64 | # Load data, initialize session |
| 65 | config=tf.ConfigProto() |