原创 編程速記(8):Python篇-獲取文件夾下指定文件後綴名的文件名

basic:os.listdir() & str.endswith() import os def get_list_img(data_dir): files = os.listdir(data_dir) basic_pa

原创 編程速記(12):保存日誌-保存print()函數輸出的訓練過程

一、場景描述 很多時候,利用服務器run model的時候,我們會選擇逐epoch打印一些數據信息(比如loss)以便分析模型訓練情況。 這個時候,保存這些訓練日誌就變得很重要了。尤其是在ubuntu服務器上利用tmux進行分屏操

原创 編程速記(7):Pytorch篇-報錯:Invalid device ids - 多GPU並行

報錯 invalid device ids 場景 在運行多GPU的pytorch程序時,報該錯誤 原因 因爲pytorch 默認是隻有cuda 0 是可見的。可以運行下面代碼: import torch print(torc

原创 編程速記(10):Pytorch篇-detach()與tensor轉numpy

報錯 原因分析 涉及到梯度傳播的tensor變量不能夠直接numpy()轉numpy,需要使用: var.detach().numpy()

原创 編程速記(17):numpy篇-對np.array做閾值限定

一、場景描述 如果我們希望對一個np.array做閾值限制,即令大於某一個數的元素賦值爲a,否則則賦值爲b 二、demo import numpy as np a = np.asarray([1,2,3]) print(a>1)

原创 編程速記(18):Python篇-引入上級目錄中模塊

demo 例如希望引入當前main.py的上級目錄中的utils文件夾下所有py文件 一種錯誤的做法是: from ../utils import * 正確的做法是: import sys sys.path.append("..

原创 編程速記(11):Python篇-圖像讀取-skimage.io.imsave&imread

一、圖像類型與通道channel 圖像類型 通道數 灰度圖像 1 彩色圖像 3(RGB) 經過測試,經過x.png格式應該是RGBA四通道,但是使用imread讀取出來的矩陣仍然是三通道。測試代碼如下: #

原创 科研速記(7):對抗樣本篇-ICCV19-Sparse and Imperceivable Adversarial Attacks

一、摘要 Motivation 與pixel-wise 的擾動相比,高度稀疏的對抗攻擊是更加危險的,因爲更加不易被偵測到。 因此,本文的研究中心在於最小化對抗樣本與乾淨樣本之間的l0l_0l0​距離。 簡述本文工作 in

原创 (待續)科研速記(1):圖像去噪篇ICCV19-Fully Convolutional Pixel Adaptive Image Denoiser

一、概覽 面向的問題 貢獻 數據集 論文實驗效果 復現實驗效果 FC-AIDE 自適應圖像去噪 1.擴展了AIDE的框架,該框架將去噪器表達爲基於上下文的像素級別的映射2.使用全卷積增強基礎的監督模型3.引入正則化

原创 編程速記(9):Python篇-字典與JSON字符串的相關操作

一、創建字典 dic = dict() # 創建空字典 二、字典插入元素 dic.update(key=value) 注:比較頭疼的是key在這裏只能是常量而不能是變量 三、獲取字典key的數量 len(dic.keys())

原创 編程速記(19):Git篇-忽略提交 .gitignore

一、場景描述 很多時候,或者是因爲項目中存在大文件無法成功提交到遠程倉庫(比如數據集),或者有某些不希望提交的文件等,就需要使用git的忽略提交 二、編寫.gitignore文件 2.1.語法 匹配目錄 使用功能斜槓"/"開頭

原创 科研速記(2):ICCV19-Wavelet Domain Style Transfer for an Effective Perception-distortion Tradeoff

Zeros Paper:Wavelet Domain Style Transfer for an Effective Perception-distortion Tradeoff in Single Image Super-Res

原创 編程速記(13):Python篇-基於argparse包傳遞命令行參數

一、場景描述 假定我們爲model定義了以下設置 在服務器的環境下,一個常見的操作是希望設置其中的變量不同的值來run該程序,再進一步分析這個變量對於我們的model的性能的影響。如果我們希望同時對某一程序run不同的進程並

原创 編程速記(21):Shell語法篇-字符串中$的使用與反斜槓

一、場景描述 在bash文件中,希望將某個程序的輸出保存到一個文件,文件名字爲attack_$attack_method$method.txt 在這個場景中,涉及到多個變量的取值,而且在字符串中我們也需要顯式指出一個變量到哪裏爲止

原创 編程速記(15):Python篇-assert斷言的應用

一、場景描述 我們希望檢測一類條件,如果條件不滿足就中斷程序運行,並且可以給出錯誤信息描述。一類簡單的解決方案就是採用斷言。 二、Demo assert condition, "error info" 例如: assert