原创 類別不均衡處理辦法總結
對於深度學習而言,數據量不平衡是很常見的問題,如:工廠的產品缺陷率一般在 0.1% 左右,患者病例比較少等。最近遇到個項目,樣本類別極其不均衡,多的類別有上萬張,少的僅有幾百張,爲了有好的訓練效果,嘗試了很多方法,在這篇博客總結下我
原创 深度學習 warmup 策略
一、介紹 warmup顧名思義就是熱身,在剛剛開始訓練時以很小的學習率進行訓練,使得網絡熟悉數據,隨着訓練的進行學習率慢慢變大,到了一定程度,以設置的初始學習率進行訓練,接着過了一些inter後,學習率再慢慢變小;學習率變化:上
原创 計算機視覺中注意力機制基礎知識(Attention Mechanism)
最近學習了關於計算機視覺中的注意力機制一些基礎知識,整理下,方便複習,也分享一下; 一、前言 二、分類 也就是兩類;軟注意力與強注意力,如下 軟注意力: 爲了更清楚地介紹計算機視覺中的注意力
原创 Tensorflow訓練出現could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 錯誤
Tensorflow1.10 Gpu:2080 cuda:9 問題: 執行訓練的時候,就算batchsize設置爲1,也會出現一下問題 could not create cudnn handle: CUDNN_STAT
原创 ubuntu系統boot空間不足解決方案(No space left on device)
問題: 執行更新命令apt-get update時,出現No space left on device: cannot copy extracted data for './boot/vmlinuz-4.4.0-116-generic'
原创 YOLOV3+ASFF(Learning Spatial Fusion for Single-Shot Object Detection)訓練踩坑記
文章:https://arxiv.org/pdf/1911.09516v2.pdf github:https://github.com/ruinmessi/ASFF 博客分析:https://blog.csdn.net/weixin_42
原创 獲取網絡模型的每一層參數量與計算量(Flops)———Pytorch
一、前言 在現在AI各種技術需要luo地的時期,網絡模型大小能夠滿足嵌入式平臺極爲重要,不僅僅需要看模型效果,也要看模型的計算量與參數,所以在評估模型的時候就要分析網絡的參數量與計算量; 二、推薦pytorch工具
原创 Ubuntu16.04系統安裝QQ+百度雲盤+微信+迅雷下載等
重裝系統後,有很多必要的軟件,博主目前常見的軟件安裝教程彙總在下面鏈接,需要的可以看看; Ubuntu系統軟件安裝教程系列 一、引言 對於用windows習慣的人來說,微信QQ 百度雲盤,迅雷下載,Foxmail等軟件很好用,但
原创 PyTorch 學習之Ubuntu16.04 下安裝 PyTorch(conda與pip兩種方式安裝)
一、pip安裝Pytorch 這裏已經默認已經安裝好了顯卡驅動,cuda等。我這裏安裝的是cuda9與cudnn7,顯卡驅動號:Driver Version: 384.130 ; 1、進入PyTorch官網:
原创 Pytorch:transforms-進行在線圖像增強的方法總結
本文截取自《PyTorch 模型訓練實用教程》,獲取全文pdf請點擊:https://github.com/tensor-yu/PyTorch_Tutorial 本文對transforms.py中的各個預處理方法進行介紹和總結。主要從官
原创 Ubuntu16.04系統VScode與常用插件推薦
重裝系統後,有很多必要的軟件,博主目前常見的軟件安裝教程彙總在下面鏈接,需要的可以看看; Ubuntu系統軟件安裝教程系列 VScode是我最喜歡的一個代碼編輯器,因爲佔用內存小,切ui看着舒服,最重要的是有有很多方便的插件,極大的提高的
原创 Numpy報錯解決辦法
bug描述: 經常出現加載numpy會報錯或者在安裝tensoflow、pytorch時,出現如下錯誤: 問題一: RuntimeError: module compiled against API version 0xc but th
原创 ubuntu16.04安裝網易雲音樂方法出現問題及解決方法(桌面圖標打不開、不能輸入中文等問題)
關於Ubuntu系統常用軟件安裝我建立一個分類,用來記錄菜鳥揪心的安裝歷程。。。 Ubuntu16.04系統安裝系列: Ubuntu配置TensorFlow—GPU版本 Ubuntu系統安裝搜狗輸
原创 TensorFlow學習(五)之Ubuntu16.04安裝TensorFlow———GPU版本(gtx1070+ubuntu16.04)
Ubuntu16.04系統安裝系列: Ubuntu系統安裝QQ+百度雲盤+微信+迅雷下載等 Ubuntu系統安裝網易雲音樂 Ubuntu系統安裝搜狗輸入法
原创 ubuntu16.04安裝Detectron教程及解決方案
本博客介紹瞭如何安裝Detectron,其依賴項(包括Caffe2)和COCO數據集。 安裝前所需知道的知識: a、Detectron運營商目前沒有CPU實施; 需要GPU系統。 b、caffe2 已經集成到pytorch1.0中,所以框