mirrors / dmlc--dgl

项目文件夹

文件

T

Tong He cbee427839 [Model] Scene Graph Extraction Model with GluonCV (#1260 )

* add working scripts

* add frcnn training script

* remove redundent files

* refactor validation computation, will optimize sgdet and training

* validation finally finished

* f-rcnn training

* test reldn

* rm file

* update reldn training

* data preprocess to h5

* temp

* use coco json

* fix conflict

* new obj dataset for detection

* update training

* before cleanup

* remove abundant files

* add arg parse to train

* cleanup code file

* update

* fix

* add readme

* add ipynb as demo

* add demo pic

* update readme

* add demo script

* improve paths

* improve readme

* add docstrings

* fix args description

* update readme

* add models from s3

* update README

Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>

2020-03-05 11:54:26 +08:00

5 行

327 B

Bash

原始文件 永久链接 Blame 文件历史

 MXNET_CUDNN_AUTOTUNE_DEFAULT=0 python validate_reldn.py \
     --pretrained-faster-rcnn-params faster_rcnn_resnet101_v1d_visualgenome/faster_rcnn_resnet101_v1d_custom_best.params \
     --reldn-params params_resnet101_v1d_reldn/model-8.params \
     --faster-rcnn-params params_resnet101_v1d_reldn/detector_feat.features-8.params