Google Colaboratoryでディレクトリに必要なファイルを作りたい

Question

import os 
import cv2
import numpy as np

def make_image(input_img):
  #画像のサイズ
  img_size=input_img.shape
  filter_one=np.ones((3,3))

  #回転用
  mat1=cv2.getRotationMatrix2D(tuple(np.array(input_img.shape[:2])/2),23,1)
  mat2=cv2.getRotationMatrix2D(tuple(np.array(input_img.shape[:2])/2),144,0.8)

  #水増しメソッドに使う関数です
  fake_method_array=np.array([
                              lambda image:cv2.warpAffine(image,mat1,image.shape[:2]),
                              lambda image:cv2.warpAffine(image,mat2,image.shape[:2]),
                              lambda image:cv2.threshold(image,100,255,cv2.THRESH_TOZERO)[1],
                              lambda image:cv2.GaussianBlur(image,(5,5),0),
                              lambda image:cv2.resize(cv2.resize(image,(img_size[1]//5,img_size[0]//5)),(img_size[1],img_size[0])),
                              lambda image:cv2.erode(image,filter_one),
                              lambda image:cv2.flip(image,1),
  ])
  
  #画像変換処理を行います。
  images=[]

  for method in fake_method_array:
    faked_img=method(input_img)
    images.append(faked_img)

  return images

#画像を読み込む
target_img=cv2.imread("IMG_1565.JPG")

#画像を水増しする
fake_images=make_image(target_img)

#画像を保存するフォルダを作成します。
if not os.path.exists("fake_images"):
  os.mkdir("fake_images")

for number,img in enumerate(fake_images):
    #まず保存先のディレクトリ"fake_images/"を指定して番号を付けて保存します。

   cv2.imwrite("fake_images/"+ str(number) + ".jpg",img)
----------------------------------------------------------------------------------------------------------------------------------------------
ls fake_images
0.jpg  1.jpg  2.jpg  3.jpg  4.jpg  5.jpg  6.jpg
----------------------------------------------------------------------------------------------------------------------------------------------
import numpy as np
import cv2
import matplotlib.pyplot as plt
import os

#列の数を設定します
NUM_COLUMNS=4
#行
ROWS_COUNT=len(fake_images)%NUM_COLUMNS
#列
COLUMS_COUNT=NUM_COLUMNS

#グラフオブジェクト保存用
subfig=[]
#figureオブジェクト作成サイズを決めます
fig=plt.figure(figsize=(12,9))

#
for i in range(1,len(fake_images)+1):
  #順序i番目のsubfig追加
  subfig.append(fig.add_subplot(ROWS_COUNT,COLUMS_COUNT,i))
  
  print(os.listdir('fake_images/'))
  print(os.path.abspath('fake_images/'+str(i-1)+'.jpg'))
  img_bgr=cv2.imread('fake_images/'+str(i-1)+'.jpg')
  img_rgb=cv2.cvtColor(img_bgr,cv2.COLOR_BGR2RGB)
  subfig[i-1].imshow(img_rgb)

#グラフ間の横と縦の隙間の調整
fig.subplots_adjust(wspace=0.3,hspace=0.3)

plt.show()


/content/fake_images/0.jpg
---------------------------------------------------------------------------
error                                     Traceback (most recent call last)
<ipython-input-39-2a2a9073e3b0> in <module>()
     24   print(os.path.abspath('fake_images/'+str(i-1)+'.jpg'))
     25   img_bgr=cv2.imread('fake_images/'+str(i-1)+'.jpg')
---> 26   img_rgb=cv2.cvtColor(img_bgr,cv2.COLOR_BGR2RGB)
     27   subfig[i-1].imshow(img_rgb)
     28 

error: OpenCV(4.1.2) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

まず、一つの画像に輪郭抽出や回転などの処理を加えて、水増しのコードを実行しました。上記のコードを実行したとき、fake_imagesというディレクトリに7つの画像が入っているはずだったのですが、結果としてfake_imagesというディレクトリには画像ファイルが入っていないということがわかりました。上記で示した通り、[ls fake_images]で確認したときは、7つのファイルがあると表示されるのですが、確認してみるとディレクトリには1つもファイルが存在しませんでした。

Comments

[jbpb0]

質問のコードの先頭に
import numpy as np
を追加し、先頭から最初の
----------------------- (略)
の行のすぐ上の
os.mkdir("fake_images")
までを、Google Colabの新規ノートブックにコピペして実行し、次のセルに
!ls -l fake_images
と書いてそれも実行したら、
total 0
と表示されます

つまり、「fake_images」ディレクトリには何もありません

質問のコードの上記の範囲までを実行したら、「fake_images」というリストに画像データが入ってる状態にはなりますが、それをファイルに保存する部分のコードが無いので、

> fake_imagesというディレクトリに7つの画像が入っているはず

とはなりません

https://teratail.com/questions/355381
の質問のコードの最後の
for number,img in enumerate(fake_images):
cv2.imwrite("fake_images/"+ str(number) + ".jpg",img)
を付けたら画像ファイルができるのに、何で今回は省いたのですか？

ただし、上記2行はインデントが間違ってます
for number,img in enumerate(fake_images):
の行はインデント無しにしないといけません

[jbpb0]

この質問のコードの後半は、下記を変更する必要があります

変更1
img_bgr=cv2.imread('fake_images'+str(i-1)+'.jpg')
↓ 「/」を付ける
img_bgr=cv2.imread('fake_images/'+str(i-1)+'.jpg')

変更2
plt.show()
のインデントを無くす

上記のように変更して、コードの前半と後半を同じノートブックで連続して実行すれば、コード前半で作成した7個の画像ファイルを、コード後半で読み込んで全部表示できます
(実行して確認済み)

[k.nirasawa]

コードを修正しましたが、同様のエラーが出てしまいました。

[k.nirasawa]

for number,img in enumerate(fake_images):
cv2.imwrite("fake_images/"+ str(number) + ".jpg",img)
を付けたら画像ファイルができるのに、何で今回は省いたのですか？
＞これはコピーし忘れていただけで、消したわけではありません。申し訳ありませんでした。

[jbpb0]

当方で実行して確認したコードを回答に書きました

> コードを修正しましたが、同様のエラーが出てしまいました。

どこかにまだエラーの原因が残ってるので、回答のコードを参考にして、それを探してください

Answer 1

下記のコードをGoogle Colabで実行しても、エラーは出ずに正常に実行できました

import numpy as np
import os 
import cv2

def make_image(input_img):
  #画像のサイズ
  img_size=input_img.shape
  filter_one=np.ones((3,3))

  #回転用
  mat1=cv2.getRotationMatrix2D(tuple(np.array(input_img.shape[:2])/2),23,1)
  mat2=cv2.getRotationMatrix2D(tuple(np.array(input_img.shape[:2])/2),144,0.8)

  #水増しメソッドに使う関数です
  fake_method_array=np.array([
                              lambda image:cv2.warpAffine(image,mat1,image.shape[:2]),
                              lambda image:cv2.warpAffine(image,mat2,image.shape[:2]),
                              lambda image:cv2.threshold(image,100,255,cv2.THRESH_TOZERO)[1],
                              lambda image:cv2.GaussianBlur(image,(5,5),0),
                              lambda image:cv2.resize(cv2.resize(image,(img_size[1]//5,img_size[0]//5)),(img_size[1],img_size[0])),
                              lambda image:cv2.erode(image,filter_one),
                              lambda image:cv2.flip(image,1),
  ])

  #画像変換処理を行います。
  images=[]

  for method in fake_method_array:
    faked_img=method(input_img)
    images.append(faked_img)

  return images

#画像を読み込む
target_img=cv2.imread("IMG_1565.jpg")

#画像を水増しする
fake_images=make_image(target_img)

#画像を保存するフォルダを作成します。
if not os.path.exists("fake_images"):
  os.mkdir("fake_images")

for number,img in enumerate(fake_images):
    #まず保存先のディレクトリ"fake_images/"を指定して番号を付けて保存します。

  cv2.imwrite("fake_images/"+ str(number) + ".jpg",img)

#####
#画像が保存できてるか確認
!ls -l fake_images
#####

#import numpy as np
#import cv2
import matplotlib.pyplot as plt
#import os

#列の数を設定します
NUM_COLUMNS=4
#行
ROWS_COUNT=len(fake_images)%NUM_COLUMNS
#列
COLUMS_COUNT=NUM_COLUMNS

#グラフオブジェクト保存用
subfig=[]
#figureオブジェクト作成サイズを決めます
fig=plt.figure(figsize=(12,9))

#
for i in range(1,len(fake_images)+1):
  #順序i番目のsubfig追加
  subfig.append(fig.add_subplot(ROWS_COUNT,COLUMS_COUNT,i))

  print(os.listdir('fake_images/'))
  print(os.path.abspath('fake_images/'+str(i-1)+'.jpg'))
  #img_bgr=cv2.imread('fake_images'+str(i-1)+'.jpg')
  img_bgr=cv2.imread('fake_images/'+str(i-1)+'.jpg')
  img_rgb=cv2.cvtColor(img_bgr,cv2.COLOR_BGR2RGB)
  subfig[i-1].imshow(img_rgb)

  #グラフ間の横と縦の隙間の調整
  fig.subplots_adjust(wspace=0.3,hspace=0.3)

plt.show()

上記のコード全体を、Google Colabの新規ノートブックの一つのセルにコピペして、実行してみてください

ただし、「IMG_1565.jpg」という画像ファイルが、Google Colab上の「/content」ディレクトリに存在する必要があります

Comments

[k.nirasawa]

実行してみたところ、やっと出したい画像が表示できました。
ありがとうございました。