Begin reorganization of code.

train/inference/README.md → inference/README.md

@@ -1,7 +1,7 @@
 # Deep Object Pose Estimation (DOPE) - Inference 
 
-This repo contains a simplified version of the **Inference** script for DOPE.
-The original repo for DOPE [can be found here](https://github.com/NVlabs/Deep_Object_Pose). 
+This directory contains a simple example of inference for DOPE.
+
 
 ## Setup
 

train/inference/inference.py → inference/inference.py

@@ -1,18 +1,12 @@
 #!/usr/bin/env python
 
-
-"""
-This file starts a ROS node to run DOPE, 
-listening to an image topic and publishing poses.
-"""
-
 import cv2
 import numpy as np
 from PIL import Image
 
 import sys
 
-sys.path.append("inference")
+sys.path.append("../common/")
 
 from cuboid import Cuboid3d
 from cuboid_pnp_solver import CuboidPNPSolver

requirements.txt

@@ -1,8 +1,74 @@
-pyrr==0.10.3
-torch==1.6.0
-torchvision==0.7.0
-numpy==1.17.4
-scipy==1.5.2
-opencv_python==4.4.0.44
-Pillow==8.1.1
+torch==2.1.2
+torchvision
+tensorboardX
+boto3
+albumentations
+botocore
+certifi
+charset-normalizer
 configparser==5.0.0
+idna
+imageio
+jmespath
+joblib
+networkx
+numpy
+opencv-python-headless==4.6.0.66
+packaging
+Pillow
+protobuf==3.20.1
+pyparsing
+pyrr==0.10.3
+python-dateutil
+PyWavelets
+PyYAML
+qudida
+requests
+s3transfer
+scikit-image
+scikit-learn
+scipy
+simplejson
+six
+threadpoolctl
+tifffile
+typing_extensions
+urllib3
+
+
+## If running into dependency issues, install the version-specific requirements below in a virtual environment
+# albumentations==1.2.1
+# boto3==1.24.58
+# botocore==1.27.58
+# certifi==2022.6.15
+# charset-normalizer==2.1.1
+# idna==3.3
+# imageio==2.21.1
+# jmespath==1.0.1
+# joblib==1.1.0
+# networkx==2.8.6
+# numpy==1.23.2
+# opencv-python-headless==4.6.0.66
+# packaging==21.3
+# Pillow==9.2.0
+# protobuf==3.20.1
+# pyparsing==3.0.9
+# pyrr==0.10.3
+# python-dateutil==2.8.2
+# PyWavelets==1.3.0
+# PyYAML==6.0
+# qudida==0.0.4
+# requests==2.28.1
+# s3transfer==0.6.0
+# scikit-image==0.19.3
+# scikit-learn==1.1.2
+# scipy==1.9.0
+# simplejson==3.17.6
+# six==1.16.0
+# tensorboardX==2.5.1
+# threadpoolctl==3.1.0
+# tifffile==2022.8.12
+# torch==1.12.1
+# torchvision==0.13.1
+# typing_extensions==4.3.0
+# urllib3==1.26.12