3D-R2N2代码解读
一、运行demo.py
文件进行预测
python demo.py prediction.obj
#前面是运行的文件,后面是将三维模型保存到这个命名的文件中
将重建出来的三维模型保存到prediction.obj
文件中
一、demo.py
代码解读
1、主函数分析
- 首先判断执行脚本的参数,确定将三维重建模型保存的名字
pred_file_name = sys.argv[1] if len(sys.argv) > 1 else 'prediction.obj'
len(sys.argv)
这个代码计算执行命令的参数
sys.argv[1]
表示执行命令行的第二个参数
sys.argv[0]
表示执行命令行中的脚本本身
在这个程序中:sys.argv[0]
是demo.py
,sys.argv[1]
是prediction.obj
2、导入图片
函数写在load_demo_images()
中
图片文件存在/imgs
中,图片的名称是数字.png
形式,如果需要改变,就在函数中进行更改
def load_demo_images():
ims = []#列表
for i in range(3):
im = Image.open('imgs/%d.png' % i)#图片的存放地点:/img/number.png
ims.append([