In order to run the code, the following libraries are required:
- python >= 3.7
- PyTorch >= 1.6
- detectron2 >= 0.2.1
- cv2 >= 4.4.0
- scikit-learn >= 0.23.2
- imantics
- easydict
Note: Please make sure the detectron2 version correctly corresponds to the pytorch and cuda versions installed. Please check this page for installation instructions and common issues.