Revision 8409361b704cd2e90b410a71165a6073caa8f13a authored by Brett Zachary Allen on 23 February 2021, 23:53:35 UTC, committed by Facebook GitHub Bot on 23 February 2021, 23:54:53 UTC
Summary: Add a feature extraction script that uses Huggingface's Fast RCNN structure and code to discover features in an image. `extract_features_frcnn.py` is my script, the other files were taken from Huggingface, other than super minor changes by me, including rotating the RGB/BGR values in the preprocessing script. The script takes a number of flags that alter it's behavior like exclude files, input files, output files, batch size, etc. Run `python3 mmf/tools/scripts/features/frcnn/extract_features_frcnn.py --model_file model.bin --config_file config.yaml --image_dir ./example_images --output_folder ./output_features` Pull Request resolved: https://github.com/facebookresearch/mmf/pull/764 Reviewed By: vedanuj Differential Revision: D26612531 Pulled By: brettallenyo fbshipit-source-id: aed9e127d9e102b2704cadd131e80bd594bce28e
1 parent f5ff2c8
| File | Mode | Size |
|---|---|---|
| ban | ||
| butd | ||
| hateful_memes | ||
| lorra | ||
| lxmert | ||
| m4c | ||
| m4c_captioner | ||
| mmbt | ||
| mmf_transformer | ||
| movie_mcan | ||
| others | ||
| pretrain_vl_right | ||
| pythia | ||
| vilbert | ||
| visual_bert |
Computing file changes ...
