图生360度部署 Panodiff

目录

Panodiff

算法步骤:

[1. python build_dataset.py crop_num pairs_num raw_data_path](#1. python build_dataset.py crop_num pairs_num raw_data_path)

[2. Relative Pose Prediction](#2. Relative Pose Prediction)

报错:

[No module named 'pytorch_lightning.utilities.distributed'](#No module named 'pytorch_lightning.utilities.distributed')

[NotImplementedError: No operator found for `memory_efficient_attention_forward` with inputs:](#NotImplementedError: No operator found for memory_efficient_attention_forward with inputs:)


Panodiff

GitHub - shanemankiw/Panodiff: Official implementation for ACM MM 2023 paper '360-Degree Panorama Generation from Few Unregistered NFoV Images'

模型和数据都在提供的压缩包里面。

算法步骤:

1. python build_dataset.py crop_num pairs_num $raw_data_path
2. Relative Pose Prediction
bash 复制代码
GPU=... # Define your GPU Device
save_path=data/pred_results.npy # Define the path for saved predictions

python pred_rotations.py \
    config.yaml \
    --classification_model_path=ckpts/stage1_classification_ckpt.pt \
    --overlap_regression_model_path=ckpts/stage2_overlap_ckpt.pt \
    --nonoverlap_regression_model_path=ckpts/stage2_nonoverlap_ckpt.pt \
    --gpu=$GPU \
    --save_path=$save_path

报错:

No module named 'pytorch_lightning.utilities.distributed'

解决方法:pip install pytorch-lightning==1.5.0

NotImplementedError: No operator found for `memory_efficient_attention_forward` with inputs:

解决方法:

NotImplementedError: No operator found for `memory_efficient_attention_forward` with inputs:-CSDN博客

相关推荐
晓数2 小时前
【硬核干货】JetBrains AI Assistant 干货笔记
人工智能·笔记·jetbrains·ai assistant
我的golang之路果然有问题2 小时前
速成GO访问sql,个人笔记
经验分享·笔记·后端·sql·golang·go·database
lwewan2 小时前
26考研——存储系统(3)
c语言·笔记·考研
搞机小能手3 小时前
六个能够白嫖学习资料的网站
笔记·学习·分类
nongcunqq3 小时前
爬虫练习 js 逆向
笔记·爬虫
汐汐咯3 小时前
终端运行java出现???
笔记
无敌小茶5 小时前
Linux学习笔记之环境变量
linux·笔记
帅云毅6 小时前
Web3.0的认知补充(去中心化)
笔记·学习·web3·去中心化·区块链
豆豆6 小时前
day32 学习笔记
图像处理·笔记·opencv·学习·计算机视觉
nenchoumi31196 小时前
VLA 论文精读(十六)FP3: A 3D Foundation Policy for Robotic Manipulation
论文阅读·人工智能·笔记·学习·vln