Install the Backend
sudo apt-get install tk-dev python-tk python3-tk
Install Matplotlib
pip install matplotlib
from pathlib import Path | |
import cv2 | |
import numpy as np | |
import depthai as dai | |
if __name__ == "__main__": | |
model_path = Path(__file__).parent / 'custom_ops/out' | |
pipeline = dai.Pipeline() |
import numpy as np | |
from fastai2.vision.all import * | |
from fastai2.distributed import * | |
def train(): | |
path = untar_data(URLs.CAMVID_TINY) | |
def label_func(fn): | |
return path/"labels"/f"{fn.stem}_P{fn.suffix}" |
#!/usr/bin/env python3 | |
''' | |
Scripts to train a keras model using tensorflow. | |
Basic usage should feel familiar: python train_v2.py --model models/mypilot | |
Usage: | |
train.py [--tubs=tub1,tub2] (--model=<model>) [--type=(linear|inferred|tensorrt_linear|tflite_linear)] | |
Options: | |
-h --help Show this screen. |
#VEHICLE | |
DRIVE_LOOP_HZ = 50 # the vehicle loop will pause if faster than this speed. | |
MAX_LOOPS = None # the vehicle loop can abort after this many iterations, when given a positive integer. | |
# #CAMERA | |
CAMERA_TYPE = "LEOPARD" # (PICAM|WEBCAM|CVCAM|CSIC|V4L|MOCK|LEOPARD) | |
IMAGE_W = 224 | |
IMAGE_H = 224 | |
IMAGE_DEPTH = 3 # default RGB=3, make 1 for mono | |
CAMERA_FRAMERATE = DRIVE_LOOP_HZ |
#VEHICLE | |
DRIVE_LOOP_HZ = 50 | |
#CAMERA | |
CAMERA_TYPE = "LEOPARD" | |
IMAGE_W = 224 | |
IMAGE_H = 224 | |
IMAGE_DEPTH = 3 | |
CAMERA_FRAMERATE = DRIVE_LOOP_HZ | |
CAMERA_VFLIP = False |
sudo apt-get install tk-dev python-tk python3-tk
pip install matplotlib
Make sure you set -D BUILD_opencv_cudacodec=OFF
.
cmake -D CMAKE_BUILD_TYPE=RELEASE \
-D WITH_CUDA=ON \
-D CUDA_ARCH_PTX="" \
-D BUILD_opencv_cudacodec=OFF \ #
-D CMAKE_INSTALL_PREFIX=/usr/local \
nvcc
is on $PATH
. Add the following lines to your ~/.bashrc
file.# Add this to your .bashrc file
export CUDA_HOME=/usr/local/cuda
# Adds the CUDA compiler to the PATH
export PATH=$CUDA_HOME/bin:$PATH
# Adds the libraries
# If not running interactively, don't do anything | |
case $- in | |
*i*) ;; | |
*) return;; | |
esac | |
# don't put duplicate lines or lines starting with space in the history. | |
# See bash(1) for more options | |
HISTCONTROL=ignoreboth |