原创 ValueError：The passed save_path is not a valid checkpoint:./experiment_VGG/model-500000
錯誤提示： ValueError：The passed save_path is not a valid checkpoint: ./experiment_VGG/model-500000 三種可能的錯誤解決： 1、注意 "/
2020-07-02 22:53:49
原创 python 尋找文件夾下含有相同字符的所有文件名
最近在做一個項目時，需要把某文件夾下的含有某些相同字符的文件的文件名字讀取並生成一個列表文件：如圖尋找下面文件夾中全部的含“traindata_017”的所有文件名字。 代碼： def readTfrecordNames(halfFi
2020-07-02 22:53:49
原创 python 將文件夾中的部分圖片轉移到另一文件夾
隨機地將文件夾中的部分圖片轉移到另一文件夾，最近在做一個網絡模型的訓練，發現測試集裏的圖片相對有點多，於是想按類別分別將測試集對應類別文件夾下的圖片隨機選取一半轉移到訓練集對應的類別文件夾下，但幾百個類別，手動轉移太麻煩，於是寫了一段代碼
2020-07-02 22:53:49
原创 SRGAN生成器Deconv的pixelShuffler代碼理解
假設input是 [b_s,h,w,256] 形狀的數組(batch_size簡寫爲b_s)： def pixelShuffler(inputs, scale=2): size = tf.shape(inputs) # siz
2020-07-02 22:53:39
2
原创 pytorch 數據處理
# -*- coding: utf-8 -*- import argparse import os import copy import torch from torch import nn import torch.optim as
2020-06-16 14:13:16
1
原创 DataLoader 加載數據並顯示數據
# -*- coding: utf-8 -*- import argparse import os import copy import torch from torch import nn import torch.optim as
2020-06-03 15:58:17
原创 nn.Sigmoid torch
from torch import nn import torch test = torch.tensor([1, 5., 4, 8, 9]) s = nn.Sigmoid() print(s(test)) tes
2020-06-02 08:42:00
原创 文件操作 python
文件操作：r(只讀)、w(只寫)、a（追加） #讀取當前目錄下的文件rnn.py f = open("rnn.py", "r",encoding="utf-8") #沒有"r"，默認也是讀模式 d=f.read() #因爲是"r"，所
2020-06-02 08:42:00
原创 圖片結構相似性及python核心代碼實現
圖片結構相似度比較公式： """X,Y是輸入的兩張圖片 """ ndim = X.ndim # ndimage filters need floating point data X = X.astype(np.float64) Y
2020-06-02 08:42:00
原创 pytorch GAN
from __future__ import print_function #%matplotlib inline import argparse import os import random import torch import
2020-06-02 08:42:00
原创 torchvision.models.vgg19獲取中間層的輸出
import math import torch from torch import nn from torchvision import models model = models.vgg19(pretrained=False) m
2020-05-24 18:13:57
原创 pytorch 關於[..., n]
input = torch.randn(1, 3, 28, 36) print(input.size()) print(input[..., 0].shape) print(input[..., 1].shape) print(inpu
2020-05-22 22:59:54
原创 tensorflow 基本用法
# 構建圖 # 1、構建圖的第一步, 是創建源 op (source op). 源 op 不需要任何輸入, 例如 常量 (Constant). 源 op 的輸出被 # 傳遞給其它 op 做運算. # 2、Python 庫中, op 構造
2020-05-22 22:59:54
原创 保存和加載模型 pytorch
# 保存和加載模型 # 當保存和加載模型時，需要熟悉三個核心功能： # 1、torch.save：將序列化對象保存到磁盤。此函數使用Python的pickle模塊進行序列化。使用此函數可以保存如模型、tensor、 # 字典等各種對象。
2020-05-22 22:59:54
原创 pytorch 卷積權重形狀
# -*- coding: utf-8 -*- import argparse import os import copy import torch from torch import nn import torch.optim as
2020-05-22 22:59:54