Retraining data for Object Detection

siddhusiddhartha25 · January 29, 2021, 6:31am

Sir, I have done annotations on some frames. How can i use those annotations to retrain the model.

AastaLLL · January 29, 2021, 7:29am

Hi,

Please find this tutorial for the retraining details:

dusty-nv/jetson-inference/blob/master/docs/pytorch-transfer-learning.md

<img src="https://github.com/dusty-nv/jetson-inference/raw/master/docs/images/deep-vision-header.jpg" width="100%">
<p align="right"><sup><a href="depthnet.md">Back</a> | <a href="pytorch-cat-dog.md">Next</a> | </sup><a href="../README.md#hello-ai-world"><sup>Contents</sup></a>
<br/>
<sup>Transfer Learning</sup></s></p>

# Transfer Learning with PyTorch

Transfer learning is a technique for re-training a DNN model on a new dataset, which takes less time than training a network from scratch.  With transfer learning, the weights of a pre-trained model are fine-tuned to classify a customized dataset.  In these examples, we'll be using the <a href="https://arxiv.org/abs/1512.03385">ResNet-18</a> and [SSD-Mobilenet](pytorch-ssd.md) networks, although you can experiment with other networks too.

<p align="center"><a href="https://arxiv.org/abs/1512.03385"><img src="https://github.com/dusty-nv/jetson-inference/raw/python/docs/images/pytorch-resnet-18.png" width="600"></a></p>

Although training is typically performed on a PC, server, or cloud instance with discrete GPU(s) due to the often large datasets used and the associated computational demands, by using transfer learning we're able to re-train various networks onboard Jetson to get started with training and deploying our own DNN models.  

<a href=https://pytorch.org/>PyTorch</a> is the machine learning framework that we'll be using, and example datasets along with training scripts are provided to use below, in addition to a camera-based tool for collecting and labeling your own training datasets.  

## Installing PyTorch

If you are [Running the Docker Container](aux-docker.md) or optionally chose to install PyTorch back when you [Built the Project](building-repo-2.md#installing-pytorch), it should already be installed on your Jetson to use.  Otherwise, if you aren't using the container and want to proceed with transfer learning, you can install it now:

``` bash

This file has been truncated. show original

Thanks.

Topic		Replies	Views
Re-trained Pytorch Mask-RCNN inferencing in Jetson Nano Jetson Nano pytorch	2	1552	October 18, 2021
Nano B01 crashes while installing PyTorch Frameworks pytorch	1	1106	June 21, 2020
Jetson Nano - Transfer Learning on Object Detection Model Jetson Nano	13	3178	October 14, 2021
Transfer Learning without TLT Jetson Xavier NX ai-training	2	631	October 18, 2021
Age and gender detection using jetson nano Jetson Nano machine-learning , ai-model-training	2	56	September 2, 2024
Hello AI World Training Cat/Dog Jetson Nano ai-training	7	463	March 29, 2024
What is the proper way to do classification pretrained? Jetson Nano ai-training	5	642	July 29, 2022
Re-training inception v4 Jetson Nano jetson-inference	4	1038	December 1, 2021
Custom segmentation in dusty-nv inference Jetson AGX Orin jetson-inference	5	763	September 21, 2022
How to using jetson inference to training my hand datasets? Jetson Nano jetson-inference	3	21	November 4, 2024

Retraining data for Object Detection

Related topics