pytorch笔记 文件名的提取(后面,7)

1.模型保存与加载

(1)保存和加载模型的参数:

保存:torch.save(the_model.state_dict(),PATH)

加载:the_model.load_state_dict(torch.load(PATH))

(2)保存和加载整个模型

torch.save(the_model,PATH)

the_model=torch.load(PATH)

2.每次做反向传播之前都要梯度置零,不然梯度会累计在一起,造成结果不收敛的情况.

此外:loss是一个Variable,所以每次要通过loss.data取出其中的Tensor,再通过loss.data[0]得到一个int或者float类型的数据,这样才能够打印出相应的数据.

3. Variable volatile=True 表示不计算梯度,默认为False , 通过该变量产生的自变量的volatile数值相同

4. detach表示隔断梯度的传播.

5. retain_graph=True,这个参数的作用:

如果说设置为False,计算图的中间变量就会在计算之后被释放掉,但是在平时使用这个参数时,默认都是False,因为为了提高代码的执行效率. 一般情况下是用不到的,但是在特殊情况中,才会遇到.如输出两个值的backward()的结果.会引发错误:

在博客中https://blog.csdn.net/wu_x_j_/article/details/84192257 有详细的例子

如:......output1.backward()#本代码正常执行,执行完中间变量都free了,所以下一个backward()会出现问题.

      output2.backward()#这个时候会发生错误.

正确写法:

.....output1.backward(retain_graph=True)

     output2.backward()

     optimizer.step() #更新参数

例如在图像的风格迁移中content loss层就会用到这个参数,因为在风格迁移中不仅仅有content loss 还有style loss层. 两个 层使用一个神经网络的参数,但是有2个loss要输出,所以要使用retain_graph=True来保留中间的参数,使得两个loss的backward()不会相互影响.

6 生成对抗网络：在训练判别器的时候，希望假的数据尽可能的输出0，训练生成器的时候，希望假的数据尽可能的输出1，定义判别器时，最后需通过一个Sigmoid函数，将结果映射为概率值，定义生成器时，最后需通过一个tanh函数，将生成的像素值映射到-1～1。

7. 对文件名的提取

代码:

import os
path_01='D:/User/wgy/workplace/data/21_training.tif'
root_01=os.path.splitext(path_01)
root_02=os.path.splitext(path_01)[0] #输出文件名
root_03=os.path.splitext(path_01)[1] #输出文件类型
print(root_01)
print(root_02)
print(root_03)
输出结果:
('D:/User/wgy/workplace/data/21_training', '.tif')  #输出文件名和文件类型
D:/User/wgy/workplace/data/21_training #文件名
.tif #文件类型

获取指定文件夹中的文件名

import os.path
rootdir = r'/home/qxq/Desktop/122/11/'  # 指明被遍历的文件夹
print(os.listdir(rootdir))

输入结果:
['123.jpg', 'dog.jpg', '1.jpg', '126.jpg', 'cat.jpg', '125.jpg', 'hellocat.jpg', '124.jpg', '111.jpg']

获取指定的文件路径什么的:

import os.path
rootdir = r'/home/qxq/Desktop/122/11/'  # 指明被遍历的文件夹
for i,j,k in os.walk(rootdir):
    print(i)
    print(j)
    print(k)

输出结果:
/home/qxq/Desktop/122/11/
[]
['123.jpg', 'dog.jpg', '1.jpg', '126.jpg', 'cat.jpg', '125.jpg', 'hellocat.jpg', '124.jpg', '111.jpg']