Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion builders/builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ def build_detection_model(name, config):
from detectors.x86_detector import X86Detector
image_width = config["image_width"]
image_height = config["image_height"]
thresh = config["threshold"]
image_thresh = config["threshold"]
detector = X86Detector(width=image_width, height=image_height, thresh=image_thresh)
else:
raise ValueError('Not supported detector named: ', name, ' for AlphaPose.')
Expand Down
6 changes: 3 additions & 3 deletions jetson.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,9 @@
# 1) build: docker build -f Dockerfile -t "neuralet/jetson-nano:tf-ssd-to-trt" .
# 2) run: docker run -it --runtime nvidia --privileged --network host -v /PATH_TO_DOCKERFILE_DIRECTORY/:/repo neuralet/jetson-nano:tf-ssd-to-trt

#FROM nvcr.io/nvidia/l4t-base:r32.4.4
#FROM nvcr.io/nvidia/l4t-base:r32.6.1

FROM nvcr.io/nvidia/l4t-tensorflow:r32.4.4-tf1.15-py3
FROM nvcr.io/nvidia/l4t-tensorflow:r32.6.1-tf1.15-py3
ENV TZ=US/Pacific
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

Expand All @@ -31,7 +31,7 @@ RUN apt-get install -y graphsurgeon-tf uff-converter-tf
RUN pip3 install protobuf
RUN apt-get install -y pkg-config libhdf5-100 libhdf5-dev
RUN apt-get install -y python3-h5py python3-opencv
RUN pip3 install wget pillow
RUN pip3 install wget pillow nvidia-pyindex

COPY ./adaptive_object_detection/exporters/libflattenconcat.so.6 /opt/libflattenconcat.so
RUN apt update && apt install -y libtcmalloc-minimal4
Expand Down