Revision 1cc6b088ed1334a5d8c197d3e6a635a6f368c3c8 authored by Ronghang Hu on 01 May 2021, 03:18:04 UTC, committed by Facebook GitHub Bot on 01 May 2021, 03:18:58 UTC
Summary:
- allow predicting attributes in Visual Genome detection -- run with `model_config.unit.predict_attributes=True` to write attributes into json prediction files
- also fix the config.yaml files for the UniT checkpoints
- visualizing predicted boxes and attributes: https://gist.github.com/ronghanghu/47053b7e58460d48986345fb9027b460#file-visualize_detection_vg_outputs_with_attr-ipynb

Pull Request resolved: https://github.com/facebookresearch/mmf/pull/910

Test Plan: tested locally and verified attribute outputs

Reviewed By: apsdehal

Differential Revision: D28101483

Pulled By: ronghanghu

fbshipit-source-id: f4db66c562d2867c473347e13db042cc9b0fd462
1 parent d30dad3
History
File Mode Size
.circleci
.github
docs
mmf
mmf_cli
projects
tests
tools
website
.editorconfig -rw-r--r-- 191 bytes
.flake8 -rw-r--r-- 187 bytes
.gitignore -rw-r--r-- 267 bytes
.pre-commit-config.yaml -rw-r--r-- 1.1 KB
LICENSE -rw-r--r-- 1.5 KB
MANIFEST.in -rw-r--r-- 130 bytes
NOTICES -rw-r--r-- 4.5 KB
README.md -rw-r--r-- 2.2 KB
pyproject.toml -rw-r--r-- 925 bytes
requirements.txt -rw-r--r-- 440 bytes
setup.py -rw-r--r-- 5.1 KB

README.md

back to top