[PyTorch] build and train a neural network model from scratch (image classification, CNN)
0. Preface
Some powerful models have been used before, such as VGG16 in the field of image classification, YoloV5 in the field of target detection, Yolact in the field of instance segmentation, etc. But only after the environment is configured for training, the interface of the source code can be modified slightly to meet their own needs. ...
Added by Rianna on Sun, 05 Dec 2021 00:33:36 +0200
caffe model to pytorch---LSTM
Before, I completed the caffe to pytorch of several networks.
refinenet https://www.cnblogs.com/yanghailin/p/13096258.html
refinedet https://www.cnblogs.com/yanghailin/p/12965695.html
The above is to extract the coffee weight and then transfer it to libtorch. The following is the conversion of the directly corresponding pytorch version to ...
Added by nahydy on Wed, 24 Nov 2021 17:25:42 +0200
PyTorch learning notes: torch.cat() and torch.stack() -- splicing of tensors
torch.cat()
torch.cat(tensors, dim=0, *, out=None) → Tensor
Official explanation: connect a given tensor sequence with a given dimension (cat stands for concatenate). All tensors must have the same shape (except the connection dimension) or be empty. It is equivalent to splicing the tensor sequence according to the specified dimension
P ...
Added by agadgil on Wed, 24 Nov 2021 03:05:25 +0200
SSD pytorch model trains its own data set
1. Download SSD pytorch code
SSD pytorch code link: https://github.com/amdegroot/ssd.pytorch
git clone https://github.com/amdegroot/ssd.pytorch
Run the code and download it locally (if the download is too slow, you can upload it to the code cloud, and then git clone the code cloud address)
2. Prepare data sets
Students without datasets ca ...
Added by jagguy on Thu, 18 Nov 2021 05:27:27 +0200
PointNet + + up sampling (Feature Propagation)
PointNet + + needs to restore the down sampled points to the same number of points as the input when processing the segmentation task, so as to facilitate the prediction of each point. But in the paper, I only give a simple description and formula, which is not very easy to understand, so I record my understanding process here.
1. Purpose of F ...
Added by Donny Bahama on Mon, 08 Nov 2021 15:14:55 +0200
Broadcasting and scientific operation of 03 tensor
Broadcasting and scientific operation of 03 tensor
1, Broadcast properties of tensors
Tensors have the same broadcast characteristics as Numpy, that is, tensors of different shapes are allowed to operate
1. Tensor calculation of the same shape
In fact, the broadcast property is also used in the operation between tensors with the same shape ...
Added by SpectralDesign.Net on Mon, 08 Nov 2021 10:47:37 +0200
Index in so net_ Function and implementation of Max
1, Background
in the Encoder part of so net classification model, first_ After pointnet gives the eigenvector, there is an operation to find the index:
M = node.size()[2]
with torch.cuda.device(self.first_pn_out.get_device()):
gather_index = index_max.forward_cuda(self.first_pn_out.detach(),
...
Added by deltawing on Thu, 04 Nov 2021 20:36:29 +0200
pytorch uses DistributedDataParallel for multi card acceleration training
stay Above We introduced how to use multithreading to accelerate model training in the data module. In this paper, we mainly introduced how to use distributed dataparallel, torch.multiprocessing and other modules in pytorch to carry out multi card parallel processing and improve the training speed of the module.The following describes pytorch's ...
Added by codersrini on Tue, 02 Nov 2021 04:24:06 +0200
Audio signal classification and recognition based on Mel spectrum (pytoch)
home pagespecial columnfront endArticle details0Audio signal classification and recognition based on Mel spectrum (pytoch)Integrity and win-win Published 5 minutes agoThis project will use pytoch to realize a simple audio signal classifier, which can be applied to mechanical signal classification and recognition, bird call signal recognition an ...
Added by throx on Fri, 29 Oct 2021 17:17:39 +0300
RPN architecture and its PyTorch implementation
introduction
because
RPN architecture
RPN
Generation method of AnchorHow to select anchor as proposalsFor loss calculation, you need to select positive and negative samples from anchor before calculating loss
In fact, RPN itself can be used as the Head of target detection
Anchor generation
So how did proposal come into being? It is ...
Added by pedroz on Mon, 25 Oct 2021 15:52:04 +0300