Googlenet training on Jetson Nano

andy8902 · July 6, 2020, 2:39am

When I use this web(jetson-inference/pytorch-collect.md at master · dusty-nv/jetson-inference · GitHub) to train googlenet , is the training googlenet’s method same as Resnet18 training on below picture? How can each category of photos be assigned to train, val, and test on googlenet training?

AastaLLL · July 6, 2020, 3:44am

Hi,

It is similar.
Just update the model information when applying training with command here:

github.com

dusty-nv/jetson-inference/blob/master/docs/pytorch-collect.md#training-your-model

<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="pytorch-plants.md">Back</a> | <a href="pytorch-ssd.md">Next</a> | </sup><a href="../README.md#hello-ai-world"><sup>Contents</sup></a>
<br/>
<sup>Transfer Learning - Classification</sup></s></p>

# Collecting your own Classification Datasets

In order to collect your own datasets for training customized models to classify objects or scenes of your choosing, we've created an easy-to-use tool called `camera-capture` for capturing and labeling images on your Jetson from live video:

<img src="https://github.com/dusty-nv/jetson-inference/raw/python/docs/images/pytorch-collection.jpg" >

The tool will create datasets with the following directory structure on disk:

```
‣ train/
	• class-A/
	• class-B/
	• ...
‣ val/
	• class-A/

This file has been truncated. show original

Thanks.