原创 Ubuntu16.04下成功安裝多版本的Cuda和切換Cuda環境 --Cuda9.0 & Cuda10.0
昨天跑個程序，報了以下錯誤： libcudart.so.9.0: cannot open shared object file: No such file or directory 我的電腦環境是 cuda10.0， torch1
2020-05-15 20:43:46
原创 Pytorch學習（三）--- 使用torchvision.transforms快速對圖像數據做數據增強
在深度學習任務中，通常讀入數據後，我們都需要對數據做transform操作，最後纔將transform後的數據送入模型進行訓練測試。 一個完整數據流pipeline可以定義爲如下： 讀取數據 -> transform -> 模型
2020-05-15 20:43:46
原创 Pytorch學習（一） --- 數據加載之Dataset類和DataLoader類
在進行深度學習任務時，一個完整的baseline通常分爲以下幾個部分： 定義模型。這裏需要構建網絡模型，後面用這個模型去訓練。 定義數據增強。這裏主要是在數據量少的情況下，對數據進行一些增強，比如平移，翻轉，裁剪等操作，以提高模
2020-05-15 20:43:46
1
原创 sudo: 在加載插件“sudoers_policy”時在 /etc/sudo.conf 第 0 行出錯 sudo: /usr/lib/sudo/sudoers.so 必須只對其所有者可寫
之前因爲修改了文件的權限爲777，導致報錯: sudo: 在加載插件“sudoers_policy”時在 /etc/sudo.conf 第 0 行出錯 sudo: /usr/lib/sudo/sudoers.so 必須只對其所有者
2020-05-15 20:43:46
原创 Pytorch學習（四） --- 模型的保存和加載
Pytorch提供了兩種方法進行模型的保存和加載。 第一種（推薦）： 該方法值保存和加載模型的參數 # 保存 torch.save(the_model.state_dict(), PATH) # 加載 # 定義模型 the_mod
2020-05-15 20:43:46
原创 模型剪枝學習筆記 --- Channel Pruning via Optimal Thresholding
Channel Pruning via Optimal Thresholding 代碼鏈接：https://github.com/yeyun11/netslim 論文鏈接：https://arxiv.org/abs/2003.04
2020-05-15 20:43:36
31
原创 PyTorch Cookbook --- 常用代碼段整理合集
本文代碼基於PyTorch 1.0版本，需要用到以下包 import collections import os import shutil import tqdm import numpy as np import PIL.I
2020-05-15 20:43:36
原创 MobileNetV3學習筆記 --- Searching for MobileNetV3
論文：https://arxiv.org/pdf/1905.02244.pdf Pytorch_MobilenetV3 實現:https://github.com/xiaolai-sqlai/mobilenetv3 給出Mobil
2020-05-15 20:43:36
原创 目標檢測中的迴歸損失函數系列二：IoU Loss
IOU Loss 出自論文：https://arxiv.org/pdf/1608.01471.pdf L1 和L2 loss是將bbox四個點分別求loss然後相加，並沒有考慮靠座標之間的相關性，而實際評價指標IOU是具備相關性。
2020-05-15 20:43:36
原创 YOLOv4真的來了！！論文翻譯 --- YOLOv4: Optimal Speed and Accuracy of Object Detection
代碼：https://github.com/AlexeyAB/darknet 論文：https://arxiv.org/abs/2004.10934 新鮮出爐的YOLOV4，集大成者。先看看效果對比： 由圖可以看到，YOLOV4
2020-05-15 20:43:36
原创 Yolact學習筆記 --- 《YOLACT Real-time Instance Segmentation》
論文：https://arxiv.org/abs/1904.02689 代碼：https://github.com/dbolya/yolact Yolact是2019年出的第一個達到實時的開創性實例分割網絡，19年底也更新到了Yo
2020-05-15 20:43:36
125
原创 YOLOv4實戰嚐鮮 --- 教你從零開始訓練自己的數據集（安全頭盔佩戴識別檢測）
目錄數據準備劃分數據集xml2yolo生成train/valid.txtYOLOv4配置cfg文件修改data文件修改names文件修改模型訓練模型測試 本文代碼基於：https://github.com/ultralytics/
2020-05-15 20:43:36
原创 Pytorch 和Tensorflow之輸入圖片數據的data_forma區別
Pytorch默認是 NCHW Tensorflow默認是NHWC N - batchsize H - 圖片高 W - 圖片寬 C - 通道數 https://blog.csdn.net/edward_zcl/article/de
2020-05-15 20:43:36
原创 Pytorch學習（二） --- 模型定義之torchvivsion.models快速構建預訓練模型
torchvision.model是torchvision一個很重要的包，裏面包含了以下模型結構： AlexNet VGG ResNet SqueezeNet DenseNet … 並且提供了預訓練模型，可以通過簡單調用來讀取
2020-05-15 20:43:36
原创 YOLOv4總結 --- 思維導圖
論文：https://arxiv.org/pdf/2004.10934.pdf 代碼：https://github.com/AlexeyAB/darknet 該圖來自公衆號【計算機視覺聯盟】
2020-05-15 20:43:36