Skip to content
This repository was archived by the owner on Jul 2, 2021. It is now read-only.

Add PSPNet model #388

Closed
wants to merge 27 commits into from
Closed
Changes from 1 commit
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
0ee2a76
Add PSPNet model
mitmul Aug 10, 2017
d31b837
Update PSPNet model
mitmul Aug 14, 2017
8c835e3
Fix convert.py
mitmul Aug 14, 2017
005c0ae
Fix URLs
mitmul Aug 15, 2017
97f3340
Merge branch 'add-cityscapes' into add-pspnet-infer
mitmul Aug 15, 2017
7787729
Revert model around auxiliary loss branch
mitmul Aug 15, 2017
163ea83
Update demo.py
mitmul Aug 15, 2017
13a5ace
Update demo.py
mitmul Aug 15, 2017
7489085
Merge branch 'master' of github.com:chainer/chainercv into add-pspnet…
mitmul Sep 25, 2017
4017ba7
Add predict.py
mitmul Oct 4, 2017
55a4f0d
update readme
mitmul Oct 4, 2017
dbe3c2b
Remove download.sh
mitmul Oct 4, 2017
07c8594
Update readme
mitmul Oct 4, 2017
e4b5ea0
Merge branch 'cityscapes-label-variable-names' of https://github.com/…
mitmul Oct 4, 2017
d40e24f
Merge branch 'add-ade20k' of https://github.com/mitmul/chainercv into…
mitmul Oct 4, 2017
78ad097
Fix
mitmul Oct 4, 2017
fbb8606
Fix import
mitmul Oct 4, 2017
891c68d
flake8
mitmul Oct 4, 2017
f567ebc
Add setup.cfg
mitmul Oct 4, 2017
1f1c382
Fix conver.py
mitmul Oct 4, 2017
0278ce5
Fix flake8
mitmul Oct 4, 2017
3f4a0b0
Merge branch 'master' of github.com:chainer/chainercv into add-pspnet…
mitmul Oct 8, 2017
d104b06
Use conv1, conv2, ... for the attribute name of ConvBNReLU link
mitmul Nov 16, 2017
09e196f
Merge DilatedResBlock and ResBlock
mitmul Nov 16, 2017
9dd625b
Always return h1 and h2 in DilatedFCN's __call__
mitmul Nov 16, 2017
a941993
Follow conventions
mitmul Dec 10, 2017
9a8d630
Merge branch 'master' of github.com:chainer/chainercv into add-pspnet…
mitmul May 18, 2018
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
Prev Previous commit
Next Next commit
flake8
mitmul committed Oct 4, 2017
commit 891c68d3029bee4d7e7a802109349bf90b22dec0
6 changes: 2 additions & 4 deletions examples/pspnet/convert.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,15 @@
import sys # NOQA # isort:skip
sys.path.insert(0, '.') # NOQA # isort:skip

import os
import re

import caffe_pb2
import chainer
import chainer.links as L
from chainer import serializers
from chainercv.links import PSPNet
from google.protobuf import text_format
import numpy as np

import caffe_pb2


def get_chainer_model(n_class, input_size, n_blocks, pyramids, mid_stride):
with chainer.using_config('train', True):