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 |
Computing file changes ...
