University/Jetbot

Object Following 공부

PrestoLive 2020. 10. 29. 09:42

1. Jetbot 기동을 통한 Object Follow

 

 

 

2. coco dataset을 사용하여 mobilenetv2를 학습 시킨 engine파일 생성(보류)

  - PyCUDA (pip install pycuda) (완료)

PyCUDA 설치

  - TensorRT 설치 (https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-zip)

 

Installation Guide :: NVIDIA Deep Learning TensorRT Documentation

These upgrade instructions are for Ubuntu and Windows users only. When upgrading from TensorRT 7.x.x to TensorRT 7.2.x, ensure you are familiar with the following. About this task Using a Debian file The Debian packages are designed to upgrade your develop

docs.nvidia.com

  - coco dataset training with mobilenetv2 (진행중)

coco dataset's classes to list
['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light', 'fire hydrant', 'street sign', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra', 'giraffe', 'hat', 'backpack', 'umbrella', 'shoe', 'eye glasses', 'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket', 'bottle', 'plate', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'mirror', 'dining table', 'window', 'desk', 'toilet', 'door', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'blender', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush', 'hair brush'