Skip to content

MaximilianWagner/medical-screw-detector

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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:

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>

Test and Deploy

Use the built-in continuous integration in GitLab.


Acknowledgment

Show your appreciation to those who have contributed to the project.

License

For open source projects, say how it is licensed.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published