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 |
---|---|---|
.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-- | 255 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-- | 2.7 KB |
README.md | -rw-r--r-- | 2.2 KB |
pyproject.toml | -rw-r--r-- | 866 bytes |
requirements.txt | -rw-r--r-- | 290 bytes |
setup.py | -rw-r--r-- | 5.0 KB |
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...