This repo compiles the submission files of the Team Forza Code; and includes the trained and inference models, test results and codebase.
To get a local copy up and running follow these simple example steps.
- Clone the repo
git clone https://github.com/HrishiMak/Summer_Challenge_ForzaCode.git
- Enter current directory
cd Summer_Challenge_ForzaCode
- Install the requirements:
pip install -r requirements.txt
To run the code, you will need to provide the paths to the following files:
- Drive Link to download checkpoints: Link
- The path to the checkpoint for the siamese network:
path_to_checkpoint1: "model_checkpoint1"
- The path to the checkpoint for the classification network:
path_to_checkpoint2: "model_checkpoint2"
- The path to the test CSV file:
path_to_test_csv: "test.csv"
- The path to the directory containing the test images:
path_to_test_imgdir: ${path_to_test_imgdir}
Once you have provided these paths, you can run the code as follows:
python inference.py --path_to_checkpoint1 ${path_to_checkpoint1} --path_to_checkpoint2 ${path_to_checkpoint2} --path_to_test_csv ${path_to_test_csv} --path_to_test_img ${path_to_test_imgdir}
This will run the inference stage and save the submission file to the current directory.
The submission file will be a CSV file with two columns:
id
: The ID of the test pairproba
: The probability that the two images in the test pair were written by the same person.