图生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博客

相关推荐
jingjingjing11116 分钟前
笔记:代码随想录算法训练营day64:拓扑排序精讲、dijkstra(朴素版)精讲
笔记
jimmyleeee10 分钟前
人工智能基础知识笔记七:随机变量的几种分布
人工智能·笔记·概率论
熬夜造bug1 小时前
LeetCode Hot100 刷题笔记(6)—— 栈、堆
笔记
对方正在长头发丿2 小时前
LETTERS(DFS)
c++·笔记·算法·深度优先·图论
能来帮帮蒟蒻吗2 小时前
GO语言学习(16)Gin后端框架
开发语言·笔记·学习·golang·gin
chushiyunen3 小时前
dom操作笔记、xml和document等
xml·java·笔记
chushiyunen3 小时前
tomcat使用笔记、启动失败但是未打印日志
java·笔记·tomcat
汇能感知3 小时前
光谱相机的光谱数据采集原理
经验分享·笔记·科技
人人题4 小时前
汽车加气站操作工考试答题模板
笔记·职场和发展·微信小程序·汽车·创业创新·学习方法·业界资讯
小脑斧爱吃鱼鱼4 小时前
鸿蒙项目笔记(1)
笔记·学习·harmonyos