Unverified Commit a585fc16 authored by Joel Shor's avatar Joel Shor Committed by GitHub
Browse files

Merge branch 'master' into master

parents fe6c3819 9d29fa9f
...@@ -9,7 +9,8 @@ research/cognitive_mapping_and_planning/* @s-gupta ...@@ -9,7 +9,8 @@ research/cognitive_mapping_and_planning/* @s-gupta
research/compression/* @nmjohn research/compression/* @nmjohn
research/delf/* @andrefaraujo research/delf/* @andrefaraujo
research/differential_privacy/* @panyx0718 research/differential_privacy/* @panyx0718
research/domain_adaptation/* @bousmalis @ddohan research/domain_adaptation/* @bousmalis @dmrd
research/gan/* @joel-shor
research/im2txt/* @cshallue research/im2txt/* @cshallue
research/inception/* @shlens @vincentvanhoucke research/inception/* @shlens @vincentvanhoucke
research/learned_optimizer/* @olganw @nirum research/learned_optimizer/* @olganw @nirum
......
...@@ -57,9 +57,8 @@ Banner () { ...@@ -57,9 +57,8 @@ Banner () {
echo -e "${green}${text}${nc}" echo -e "${green}${text}${nc}"
} }
# Download the dataset. You will be asked for an ImageNet username and password. # Download the dataset.
# To get one, register at http://www.image-net.org/. bazel build "${git_repo}/research/slim:download_and_convert_imagenet"
bazel build "${git_repo}/research/slim:download_and_convert_imagenet"
"./bazel-bin/download_and_convert_imagenet" ${DATASET_DIR} "./bazel-bin/download_and_convert_imagenet" ${DATASET_DIR}
# Run the compression model. # Run the compression model.
......
...@@ -6,7 +6,7 @@ ...@@ -6,7 +6,7 @@
"source": [ "source": [
"# TFGAN Tutorial\n", "# TFGAN Tutorial\n",
"\n", "\n",
"## Authors: joelshor@\n", "## Authors: Joel Shor, joel-shor\n",
"\n", "\n",
"### More complex examples, see [`tensorflow/models/gan`](https://github.com/tensorflow/models/tree/master/research/gan)\n", "### More complex examples, see [`tensorflow/models/gan`](https://github.com/tensorflow/models/tree/master/research/gan)\n",
"\n", "\n",
...@@ -64,7 +64,7 @@ flags.DEFINE_float('clip_gradient_norm', 0, '') ...@@ -64,7 +64,7 @@ flags.DEFINE_float('clip_gradient_norm', 0, '')
flags.DEFINE_integer('save_summaries_secs', 15, '') flags.DEFINE_integer('save_summaries_secs', 15, '')
flags.DEFINE_integer('eval_interval_secs', 60 * 5, '') flags.DEFINE_integer('eval_interval_secs', 60 * 5, '')
# Scheduling # Scheduling
flags.DEFINE_string('master', 'local', '') flags.DEFINE_string('master', '', '')
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
...@@ -44,7 +44,7 @@ flags.DEFINE_integer('image_size', 64, ...@@ -44,7 +44,7 @@ flags.DEFINE_integer('image_size', 64,
'Input images dimension (pixels) - width & height.') 'Input images dimension (pixels) - width & height.')
flags.DEFINE_integer('vox_size', 32, 'Voxel prediction dimension.') flags.DEFINE_integer('vox_size', 32, 'Voxel prediction dimension.')
flags.DEFINE_integer('step_size', 24, 'Steps to take in rotation to fetch viewpoints.') flags.DEFINE_integer('step_size', 24, 'Steps to take in rotation to fetch viewpoints.')
flags.DEFINE_integer('batch_size', 1, 'Batch size while training.') flags.DEFINE_integer('batch_size', 6, 'Batch size while training.')
flags.DEFINE_float('focal_length', 0.866, 'Focal length parameter used in perspective projection.') flags.DEFINE_float('focal_length', 0.866, 'Focal length parameter used in perspective projection.')
flags.DEFINE_float('focal_range', 1.732, 'Focal length parameter used in perspective projection.') flags.DEFINE_float('focal_range', 1.732, 'Focal length parameter used in perspective projection.')
flags.DEFINE_string('encoder_name', 'ptn_encoder', flags.DEFINE_string('encoder_name', 'ptn_encoder',
......
...@@ -45,7 +45,9 @@ ...@@ -45,7 +45,9 @@
# downloading the raw images. # downloading the raw images.
# #
# usage: # usage:
# ./download_and_convert_imagenet.sh [data-dir] # cd research/slim
# bazel build :download_and_convert_imagenet
# ./bazel-bin/download_and_convert_imagenet.sh [data-dir]
set -e set -e
if [ -z "$1" ]; then if [ -z "$1" ]; then
...@@ -58,7 +60,7 @@ DATA_DIR="${1%/}" ...@@ -58,7 +60,7 @@ DATA_DIR="${1%/}"
SCRATCH_DIR="${DATA_DIR}/raw-data/" SCRATCH_DIR="${DATA_DIR}/raw-data/"
mkdir -p "${DATA_DIR}" mkdir -p "${DATA_DIR}"
mkdir -p "${SCRATCH_DIR}" mkdir -p "${SCRATCH_DIR}"
WORK_DIR="$0.runfiles/third_party/tensorflow_models/research/slim" WORK_DIR="$0.runfiles/__main__"
# Download the ImageNet data. # Download the ImageNet data.
LABELS_FILE="${WORK_DIR}/datasets/imagenet_lsvrc_2015_synsets.txt" LABELS_FILE="${WORK_DIR}/datasets/imagenet_lsvrc_2015_synsets.txt"
......
...@@ -49,7 +49,7 @@ mkdir -p "${BBOX_DIR}" ...@@ -49,7 +49,7 @@ mkdir -p "${BBOX_DIR}"
cd "${OUTDIR}" cd "${OUTDIR}"
# Download and process all of the ImageNet bounding boxes. # Download and process all of the ImageNet bounding boxes.
BASE_URL="http://www.image-net.org/challenges/LSVRC/2012/nonpub" BASE_URL="http://www.image-net.org/challenges/LSVRC/2012/nnoupb"
# See here for details: http://www.image-net.org/download-bboxes # See here for details: http://www.image-net.org/download-bboxes
BOUNDING_BOX_ANNOTATIONS="${BASE_URL}/ILSVRC2012_bbox_train_v2.tar.gz" BOUNDING_BOX_ANNOTATIONS="${BASE_URL}/ILSVRC2012_bbox_train_v2.tar.gz"
......
...@@ -331,7 +331,7 @@ class SmallConfig(object): ...@@ -331,7 +331,7 @@ class SmallConfig(object):
lr_decay = 0.5 lr_decay = 0.5
batch_size = 20 batch_size = 20
vocab_size = 10000 vocab_size = 10000
rnn_mode = CUDNN rnn_mode = BLOCK
class MediumConfig(object): class MediumConfig(object):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment