mitmul

FROM nvidia/cuda:8.0-cudnn6-devel

RUN apt-get update -y && \
	apt-get install -y --no-install-recommends \
	python3-dev \
	python3-pip \
	python3-wheel \
	python3-setuptools \
	git \
	curl \

mitmul

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import argparse

import chainer
import chainer.functions as F
import chainer.links as L
import numpy as np

mitmul

mitmul

mitmul

import weakref
import matplotlib.pyplot as plt
import numpy
from sklearn.datasets import fetch_mldata


class Variable(object):

    def __init__(self, data):
        self.data = data

mitmul

#!/bin/bash

# Install VTK First
if [ ! -d VTK-6.3.0 ]; then
    brew uninstall vtk
    brew unisntall pcl
    brew install cmake
    brew install pkg-config
    brew install boost
    brew install cminpack

mitmul

import chainer
import chainer.functions as F
import numpy as np
import cupy as cp
import time
from chainer import cuda
import sys


if sys.argv[1] == 'cpu':

mitmul

mitmul

# git clone https://github.com/hszhao/PSPNet.git
# This is the patched version:
git clone https://github.com/mitmul/PSPNet.git
cd PSPNet && mkdir build && cd build

# cudnnenv install v4
# cudnnenv activate v4

conda install -y cmake
conda install -y lmdb

mitmul

stop_epoch: 1000
# max_workspace_size: 256

dataset:
  train:
    module: dataset.cityscapes.cityscapes
    name: TransformedCityscapes
    args:
      img_dir: data/cityscapes/leftImg8bit
      label_dir: data/cityscapes/gtFine