Automation Lab
•
2024-03-07
AI写真整理術：画像認識で自動フォルダ分け
所要時間
15 MIN
必要環境
Python 3.x, PyTorch, Pillow
1万枚以上の写真が1つのフォルダに溜まっていませんか？AIモデル（ResNet）を使って、画像の内容を解析し、カテゴリごとに自動で整理するスクリプトを紹介します。
仕組み
1. AIモデル読み込み: PyTorchを使用して学習済みモデルを起動。
2. 画像解析: 写真に写っているものが何かを判定。
3. 自動移動: 判定結果に基づいてフォルダを作成し、ファイルを移動。
実装のヒント
import torch
import torchvision.models as models
from PIL import Image
# 学習済みのResNetモデルを使用
model = models.resnet18(pretrained=True)
model.eval()
# ここで画像を読み込み、AIに予測させ、
# その結果をフォルダ名にしてファイルを移動します。
Googleフォトを使わなくても、自分のPC内で高度な写真管理が可能になります。
import torch import torchvision.models as models from PIL import Image # 学習済みのResNetモデルを使用 model = models.resnet18(pretrained=True) model.eval() # ここで画像を読み込み、AIに予測させ、 # その結果をフォルダ名にしてファイルを移動します。Googleフォトを使わなくても、自分のPC内で高度な写真管理が可能になります。