neural-motifs-master
models
train_rels.py
9KB
eval_rel_count.py
9KB
_visualize.py
9KB
eval_rels.py
4KB
train_detector.py
9KB
misc
relation_types.txt
908B
__init__.py
0B
object_types.txt
2KB
motifs.py
6KB
docs
_layouts
default.html
2KB
_config.yaml
78B
_includes
image.html
104B
upload.sh
81B
LICENSE.md
1KB
index.md
3KB
teaser.png
352KB
dataloaders
mscoco.py
7KB
image_transforms.py
4KB
__init__.py
0B
blob.py
9KB
visual_genome.py
16KB
data
stanford_filtered
README.md
1006B
LICENSE
1KB
.gitignore
1KB
lib
resnet.py
5KB
draw_rectangles
setup.py
200B
draw_rectangles.pyx
3KB
draw_rectangles.c
318KB
pytorch_misc.py
14KB
get_union_boxes.py
4KB
rel_model.py
23KB
word_vectors.py
5KB
object_detector.py
25KB
__init__.py
0B
sparse_targets.py
2KB
lstm
decoder_rnn.py
12KB
__init__.py
0B
highway_lstm_cuda
build.py
798B
src
highway_lstm_cuda.h
929B
highway_lstm_kernel.cu
18KB
highway_lstm_kernel.h
854B
highway_lstm_cuda.c
4KB
__init__.py
0B
alternating_highway_lstm.py
15KB
_ext
highway_lstm_layer
__init__.py
317B
__init__.py
0B
make.sh
2KB
surgery.py
3KB
evaluation
sg_eval.py
12KB
sg_eval_slow.py
8KB
__init__.py
0B
sg_eval_all_rel_cates.py
14KB
test_sg_eval.py
10KB
rel_model_stanford.py
9KB
fpn
generate_anchors.py
4KB
nms
functions
nms.py
1KB
build.py
814B
src
cuda
nms_kernel.h
129B
Makefile
168B
nms_kernel.cu
4KB
nms_cuda.c
567B
nms_cuda.h
85B
Makefile
139B
box_utils.py
6KB
anchor_targets.py
4KB
box_intersections_cpu
bbox.c
356KB
bbox.pyx
4KB
setup.py
178B
make.sh
297B
proposal_assignments
proposal_assignments_rel.py
9KB
proposal_assignments_postnms.py
5KB
rel_assignments.py
6KB
proposal_assignments_gtbox.py
3KB
proposal_assignments_det.py
4KB
roi_align
functions
roi_align.py
2KB
__init__.py
0B
build.py
901B
src
cuda
roi_align_kernel.cu
7KB
roi_align_kernel.h
1KB
Makefile
198B
roi_align_cuda.h
381B
roi_align_cuda.c
2KB
__init__.py
0B
Makefile
157B
_ext
__init__.py
0B
roi_align
__init__.py
308B
modules
roi_align.py
2KB
__init__.py
0B
get_dataset_counts.py
2KB
Makefile
388B
config.py
8KB
README.md
3KB
scripts
train_models_sgcls.sh
1011B
train_stanford.sh
318B
train_motifnet.sh
1KB
eval_models_sgcls.sh
2KB
eval_models_sgdet.sh
976B
pretrain_detector.sh
721B
refine_for_detection.sh
991B
暂无评论