8wDlpd.png
8wDFp9.png
8wDEOx.png
8wDMfH.png
8wDKte.png

加载 BLIP 处理器和图像字幕模型时出现问题

R13 2月前

15 0

我在使用 Salesforce/blip-image-captioning-base 模型加载 BLIP 处理器和图像字幕模型时遇到问题。我的脚本在尝试加载时似乎卡住了...

我在加载 BLIP 处理器和模型以使用该 Salesforce/blip-image-captioning-base 模型进行图像字幕制作时遇到了问题。我的脚本在尝试加载处理器和模型时似乎卡住了。以下是我的设置和我正在使用的脚本的详细信息。

环境详细信息

  • 变形金刚版本 :4.42.3
  • Torch 版本 :2.2.2
  • 枕头版本 :10.3.0
  • Python 版本 :3.12(Anaconda 环境)
  • 操作系统 :macOS
  • CUDA 可用性 :错误(在 CPU 上运行)

脚本

import logging
from transformers import BlipProcessor, BlipForConditionalGeneration, __version__ as transformers_version
from PIL import Image, __version__ as pillow_version
import torch

# Enable logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Log versions
logger.info(f"Transformers version: {transformers_version}")
logger.info(f"Torch version: {torch.__version__}")
logger.info(f"Pillow version: {pillow_version}")

# Check CUDA availability
logger.info(f"CUDA available: {torch.cuda.is_available()}")

# Load an image
image_path = "./data/images/LHorgan_JDFAirWing_279.jpg"
try:
    image = Image.open(image_path)
    logger.info("Image loaded successfully")
except Exception as e:
    logger.error(f"Failed to load image: {e}")

# Load the processor and model with detailed logging
try:
    logger.info("Loading processor...")
    processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
    logger.info("Processor loaded successfully")

    logger.info("Loading model...")
    model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
    logger.info("Model loaded successfully")
except Exception as e:
    logger.error(f"Failed to load processor or model: {e}")

# Check if the model and processor are correctly loaded
if 'processor' in locals():
    logger.info("Processor is ready for use.")
else:
    logger.error("Processor was not loaded properly.")

if 'model' in locals():
    logger.info("Model is ready for use.")
else:
    logger.error("Model was not loaded properly.")

问题

该脚本记录以下内容,然后在加载处理器时卡住:

INFO:__main__:Transformers version: 4.42.3
INFO:__main__:Torch version: 2.2.2
INFO:__main__:Pillow version: 10.3.0
INFO:__main__:CUDA available: False
INFO:__main__:Image loaded successfully
INFO:__main__:Loading processor...

它似乎无限期地挂在“正在加载处理器...”。我已确保我的互联网连接稳定,并检查了我的 Python 环境设置。尽管如此,处理器和模型似乎仍未成功加载。

有人能提供关于如何解决此问题的见解或建议吗?我可能遗漏了哪些具体配置或步骤?如能提供任何帮助,我将不胜感激。

先感谢您!

帖子版权声明 1、本帖标题:加载 BLIP 处理器和图像字幕模型时出现问题
    本站网址:http://xjnalaquan.com/
2、本网站的资源部分来源于网络,如有侵权,请联系站长进行删除处理。
3、会员发帖仅代表会员个人观点,并不代表本站赞同其观点和对其真实性负责。
4、本站一律禁止以任何方式发布或转载任何违法的相关信息,访客发现请向站长举报
5、站长邮箱:yeweds@126.com 除非注明,本帖由R13在本站《tensorflow》版块原创发布, 转载请注明出处!
最新回复 (0)
返回
作者最近主题: