YOLOv2 on Jetson TX2

I tested pre-trained YOLOv2 on Jetson TX2, and documented the result in the post.

Share "YOLOv2 on Jetson TX2"

Share on: FacebookTwitter

How to Capture Camera Video and Do Caffe Inferencing with Python on Jetson TX2

I extended my previous tegra-cam.py example by hooking up a Caffe image classification model into the video pipeline. The resulting code should be good for quickly verifying a newly trained Caffe image classification model, for prototyping, or for building Caffe demo programs with live camera input.

Share "How to Capture Camera Video and Do Caffe Inferencing with Python on Jetson TX2"

Share on: FacebookTwitter

How to Capture and Display Camera Video with Python on Jetson TX2

In this post I share how to use python code (with OpenCV) to capture and display camera video on Jetson TX2, including IP CAM, USB webcam and the Jetson onboard camera. This sample code should work on Jetson TX1 as well.

Share "How to Capture and Display Camera Video with Python on Jetson TX2"

Share on: FacebookTwitter

Training a Fish Detector with NVIDIA DetectNet (Part 2/2)

I trained a DetectNet model with the data I prepared from Kaggle's 'The Nature Conservancy Fisheries Monitoring' dataset. Here's the list of steps I employed for the training, as well as a brief discussion about the result.

Share "Training a Fish Detector with NVIDIA DetectNet (Part 2/2)"

Share on: FacebookTwitter

Training a Fish Detector with NVIDIA DetectNet (Part 1/2)

I took fish image data from Kaggle's 'The Nature Conservancy Fisheries Monitoring' competition, and wanted to train a 'fish detector' with it. I chose NVIDIA DetectNet as the underlying object detector. In this post I documented how I prepared fish image training data for DetectNet.

Share "Training a Fish Detector with NVIDIA DetectNet (Part 1/2)"

Share on: FacebookTwitter

How I Built My Own Deep Learning PC

I built my own deep learning PC for DNN/CNN training about 1 year ago. Here's how I did it.

Share "How I Built My Own Deep Learning PC"

Share on: FacebookTwitter

Trying out TensorRT on Jetson TX2

As I was learning how to use TensorRT to achieve best inference performance on Jetson TX2, I used my previously trained cats-dogs Caffe model for experiment. The result was good, as expected.

Share "Trying out TensorRT on Jetson TX2"

Share on: FacebookTwitter

Deep Learning Cats Dogs Tutorial on Jetson TX2

I ran the Deep Leanring Cats Dogs Tutorial code to train an AlexNet on Jetson TX2. This is the best Caffe and Pyhton tutorial I've come across so far. I highly recommend it to people who wants to learn Caffe.

Share "Deep Learning Cats Dogs Tutorial on Jetson TX2"

Share on: FacebookTwitter

Nintendo AI Agent Training in Action, Finally...

I finally managed to put everything together, and started training my AI Agent (DeepMind's DQN) to play 'Galaga' on Nintendo Famicom Mini. Hopefully the AI could learn in a couple of weeks to play significantly better than the random player so that I could report the progress then...

Share "Nintendo AI Agent Training in Action, Finally..."

Share on: FacebookTwitter

Letting a Random Agent Play Galaga

A random agent (player) playing Galaga can get an average score of 6314.3! Hopefully my AI agent could do much better than that after training.

Share "Letting a Random Agent Play Galaga"

Share on: FacebookTwitter

Galaga Game Environment

I integrated the vidcap, galaga and gpio modules and finalized a Galaga game environment API which is similar to xtari, the Atari game emulator.

Share "Galaga Game Environment"

Share on: FacebookTwitter

Using cuDNN to Speed Up DQN Training on Jetson TX1

I found it was quite easy to enable cudnn in Torch7. I was able to reduce DQN training time by 1/3 on Jetson TX1 with change of only a few lines of code.

Share "Using cuDNN to Speed Up DQN Training on Jetson TX1"

Share on: FacebookTwitter

Accessing Hardware GPIO in Torch7

I developd code to control GPIO outputs from Torch7. This should work for not only Jetson TX1 but other Linux platforms as well. The code could be extended to handle GPIO inputs too.

Share "Accessing Hardware GPIO in Torch7"

Share on: FacebookTwitter

Galaga (小蜜蜂)

I picked this Nintendo Famicom Mini game as my first target for the AI agent!

Share "Galaga (小蜜蜂)"

Share on: FacebookTwitter

Capturing HDMI Video in Torch7

I implemented a video capturing module in Torch7. This module calls V4L2 API (in C code) to fetch raw video data from the underlying video capture device. It then uses Lua FFI interface to pass the data to Lua/Torch7. The resulting data are Torch Tensors and could be displayed on screen with Torch7's image API.

Share "Capturing HDMI Video in Torch7"

Share on: FacebookTwitter

Nintendo Famicom Mini

Nintendo Famicom Mini, Japan's version of the NES Classic Edition.

Share "Nintendo Famicom Mini"

Share on: FacebookTwitter

Training DeepMind's DQN to Play 'Pong', the Atari Game

I made some modifications to 'DeepMind Atari Deep Q Learner' so that it could run on Jetson TX1. I then trained the DQN to play 'pong' on Jetson TX1 for a week. The result is good.

Share "Training DeepMind's DQN to Play 'Pong', the Atari Game"

Share on: FacebookTwitter

Booting Rootfs off SD Card on Jetson TX1

I build my TX1 software development environment on large SD Card. This approach has a number of benefits. I share how I build the SD Card image here.

Share "Booting Rootfs off SD Card on Jetson TX1"

Share on: FacebookTwitter

How to Install Torch7 on Jetson TX1

This article describes how I install Torch7 on Jetson TX1. Since Torch7 gets updated frequently and its building process on TX1 might break from time to time, I'll try to update this blog to include the latest necessary fixes/workarounds whenever I can.

Share "How to Install Torch7 on Jetson TX1"

Share on: FacebookTwitter

blog built using the cayman-theme by Jason Long. LICENSE