as shown in requirements.txt
- you need have installed docker and nvidia-docker.
- you should download the docker file from the google drive
- use the command to load the docker file :docker load -i ustc_nelslip_submitv1.zip
- prepare your test images and random predictions files, besides, you should prepare a "predictions" folder to contain the generated final submitted prediction file.These three parameters are necessary.
- run the code to get your final submitted prediction file in the folder "predictions" as with the following command:
sudo nvidia-docker run -ti -v $test_image_path:/test/ -v $random_predictions_path:/random_predictions.csv -v $output_predictions_path:/predictions/ ustc_nelslip_submit:v1.0 ./test/ ./predictions/
- $test_image_path -- any arbitrary directory contains input test images,like /home/xxx/ECCV/ECCV_docker/test/
- $random_predictions_path -- any arbitrary file path to the random prediction file ,like /home/xxx/ECCV/ECCV_docker/random_predictions.csv
- $output_predictions_path -- any arbitrary directory, and the final submitted prediction file would be generated in the directory. like /home/xxx/ECCV/ECCV_docker/predictions/
- prepare the orign train data, and the pretrained model from : DSFD model, IR_152 model, IR_50 model
- use the command to obtain the face from given image: python crop_face.py --old_path xx --new_path xx
- change your config with config.py, and train your model with the command: python train_demo.py
we choose two backbone: IR_50, IR_152, two head: arcface, cosface and multiple data enhancement methods, and the detailed information would be shown in a paper soon.