Medical Screw Detector

Getting started

Building custom cuda kernels

python3 setup.py build
python3 setup.py install --prefix=~/.local/

Environment

Check the environments for different setups:

• Conda Environment
• pip Cluster Environment
• pip Local Environment

Running

python main.py
  --backbone resnet101
  --dataset_reduction 10
  --lr 0.00004 
  --lr_backbone 0.000004 
  --batch_size 2 
  --epochs 47 
  --lr_drop_epochs 40 
  --data_dir <path_to_dataset>
  --result_dir <path_for_results>
  --checkpoint_file <path_to_checkpoint>.ckpt
  --backbone_checkpoint_file <path_to_backbone_checkpoint>

ResNet50 Test

srun python main.py --backbone medical_resnet50 --dataset_reduction 100 --lr 0.00004 --lr_backbone 0.000004 --batch_size 2 --epochs 10 --lr_drop_epochs 4 --with_box_refine --two_stage --eff_query_init --eff_specific_head --rho 0.1 --use_enc_aux_loss --num_queries 300 --num_workers 10 --data_dir /home/vault/iwi5/iwi5165h/V1-1to3objects-400projections-circular --result_dir /home/hpc/iwi5/iwi5165h/Screw-Detection-Results/Test-Job --backbone_checkpoint_file /home/vault/iwi5/iwi5165h/DINO-Checkpoints/checkpoint_resnet50_DINO_Training_Job_036_ResNet50_0200.pth

ViT-S-16 Test

srun python main.py --backbone medical_vit_s_16 --dataset_reduction 100 --lr 0.00004 --lr_backbone 0.000004 --batch_size 5 --epochs 10 --lr_drop_epochs 4 --with_box_refine --two_stage --eff_query_init --eff_specific_head --rho 0.1 --use_enc_aux_loss --num_queries 300 --num_workers 10 --data_dir /home/vault/iwi5/iwi5165h/V1-1to3objects-400projections-circular --result_dir /home/hpc/iwi5/iwi5165h/Screw-Detection-Results/Test-Job --backbone_checkpoint_file /home/vault/iwi5/iwi5165h/DINO-Checkpoints/checkpoint_vit_small_DINO_Training_Job_037_ViT-S-16_0200.pth