[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