In order to run the code, the following libraries are required:
- python >= 3.7
- PyTorch >= 1.6
- detectron2 >= 0.3
- cv2 >= 4.4.0
- scikit-learn >= 0.23.2
- imantics
- easydict
- h5py
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.