thomasweng15

import cv2
import rospy
import argparse
import numpy as np
import matplotlib.pyplot as plt
from cv_bridge import CvBridge
from sensor_msgs.msg import Image

class Segmenter():
    def __init__(self, args):

thomasweng15

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import TimedAnimation

class SubplotAnimation(TimedAnimation):
    def __init__(self, num_steps=50):
        fig = plt.figure()
        self.ax1 = fig.add_subplot(2, 1, 1) 
        self.ax2 = fig.add_subplot(2, 1, 2, projection='3d')
        self.ax2.set(xlim3d=(-1, 1), xlabel='X')

thomasweng15

#!/usr/bin/env python

import rospy
import numpy as np
import copy
import tf
import tf.transformations as tfs
import geometry_msgs.msg
from easy_handeye.handeye_calibration import HandeyeCalibration

thomasweng15

import torch.nn as nn
import math
import torch.utils.model_zoo as model_zoo
import numpy as np


__all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101',
           'resnet152']

thomasweng15

class ResNet(nn.Module):
    def __init__(self,
                 block,
                 layers,
                 num_classes=1000,
                 fully_conv=False,
                 remove_avg_pool_layer=False,
                 output_stride=32,
                 additional_blocks=0,
                 multi_grid=(1,1,1) ):

thomasweng15

# -*- coding: utf-8 -*-

from __future__ import print_function

from matplotlib import pyplot as plt
from PIL import Image
import numpy as np
import random
import os
import cv2

thomasweng15

import seaborn as sns; sns.set()
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

def plot(run_IDs, run_labels, factors, titles, 
         colors=['steelblue', 'orange', 'mediumseagreen', 'red', 'purple'],
         ecolors=['lightblue', 'navajowhite', 'lightgreen', 'pink', 'plum'],
         dpi=150):
    csv_names = ['run_%d/progress.csv' % i for i in run_IDs]

thomasweng15

#!/usr/bin/env python
import rospy
import numpy as np
import moveit_commander
import intera_interface
from copy import deepcopy
from geometry_msgs.msg import Pose
import sys
from visual_mpc.envs.robot_envs.sawyer.sawyer_impedance import SawyerImpedanceController
from visualization_msgs.msg import Marker

thomasweng15

#!/usr/bin/env bash
#SBATCH --nodes=1
#SBATCH --partition=GPU
#SBATCH --ntasks-per-node=1
#SBATCH --time=96:00:00
#SBATCH --gres=gpu:1
#SBATCH --mem=8G
#SBATCH --exclude=compute-0-[9,11,13]
#SBATCH -o /home/USER/out.txt
#SBATCH -e /home/USER/out.txt

thomasweng15

import rospy
import sys
from sensor_msgs.msg import Image
import pprint
import cv2
from cv_bridge import CvBridge
import tf
import time

global saved_RGB