How to run detectron2 with pycharm?

[mrtkhosravi]

I've built detectron2 on my ubuntu 20.04. I use conda as environment manager. Detectron2 is installed in conda environment.

when I use detectron in terminal it works:

(base) morty@morty-MS-7811:~/github/detectron2/tools$ conda activate detectron2
(detectron2) morty@morty-MS-7811:~/github/detectron2/tools$ ./train_net.py --config-file ../configs/COCO-
InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml --num-gpus 1 SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025

Command Line Args: Namespace(config_file='../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml', dist_url='tcp://127.0.0.1:50152', eval_only=False, machine_rank=0, num_gpus=1, num_machines=1, opts=['SOLVER.IMS_PER_BATCH', '2', 'SOLVER.BASE_LR', '0.0025'], resume=False)
[11/24 13:01:01 detectron2]: Rank of current process: 0. World size: 1
[11/24 13:01:01 detectron2]: Environment info:
----------------------  -------------------------------------------------------------------------------------
sys.platform            linux
Python                  3.8.12 (default, Oct 12 2021, 13:49:34) [GCC 7.5.0]
numpy                   1.21.2
detectron2              0.6 @/home/morty/anaconda3/envs/detectron2/lib/python3.8/site-packages/detectron2
Compiler                GCC 9.3
CUDA compiler           CUDA 11.3
detectron2 arch flags   8.6
DETECTRON2_ENV_MODULE   <not set>
PyTorch                 1.10.0 @/home/morty/anaconda3/envs/detectron2/lib/python3.8/site-packages/torch
PyTorch debug build     False
GPU available           Yes
GPU 0                   NVIDIA GeForce RTX 3060 (arch=8.6)
Driver version          470.82.00
CUDA_HOME               /usr/local/cuda-11.3
Pillow                  8.4.0

but when I run train_net.py with pycharm, it errors out. Pycharm's python console is like this:

/home/morty/anaconda3/envs/detectron2/bin/python /opt/pycharm-2021.2.3/plugins/python/helpers/pydev/pydevconsole.py --mode=client --port=37637
import sys; print('Python %s on %s' % (sys.version, sys.platform))
sys.path.extend(['/home/morty/github/detectron2'])
PyDev console: starting.
Python 3.8.12 (default, Oct 12 2021, 13:49:34) 
[GCC 7.5.0] on linux
import os
os.environ['DETECTRON2_DATASETS'] = '/home/morty/datasets'
runfile('/home/morty/github/detectron2/tools/train_net.py', args=['--config-file', '../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml', '\\', '--num-gpus', '1', 'SOLVER.IMS_PER_BATCH', '2', 'SOLVER.BASE_LR', '0.0025'], wdir='/home/morty/github/detectron2/tools')
Traceback (most recent call last):
  File "<input>", line 3, in <module>
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_bundle/pydev_umd.py", line 198, in runfile
    pydev_imports.execfile(filename, global_vars, local_vars)  # execute the script
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
  File "/home/morty/github/detectron2/tools/train_net.py", line 28, in <module>
    from detectron2.data import MetadataCatalog
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/morty/github/detectron2/detectron2/data/__init__.py", line 4, in <module>
    from .build import (
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/morty/github/detectron2/detectron2/data/build.py", line 14, in <module>
    from detectron2.structures import BoxMode
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/morty/github/detectron2/detectron2/structures/__init__.py", line 3, in <module>
    from .image_list import ImageList
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/morty/github/detectron2/detectron2/structures/image_list.py", line 8, in <module>
    from detectron2.layers.wrappers import shapes_to_tensor
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/morty/github/detectron2/detectron2/layers/__init__.py", line 3, in <module>
    from .deform_conv import DeformConv, ModulatedDeformConv
  File "/opt/pycharm-2021.2.3/plugins/python/helpers/pydev/_pydev_bundle/pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "/home/morty/github/detectron2/detectron2/layers/deform_conv.py", line 11, in <module>
    from detectron2 import _C
ImportError: cannot import name '_C' from 'detectron2' (/home/morty/github/detectron2/detectron2/__init__.py)

I tested the environment in code and it confirms the code is running in correct conda env:

import os
print('ENV:', os.environ['CONDA_DEFAULT_ENV'])

and it returns ENV: detectron2 which is the correct value.

[ChrisDAT20]

did you "build" Detectron2? Needs to be done after installing torch and torchvision.
From https://detectron2.readthedocs.io/en/latest/tutorials/install.html

python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
# (add --user if you don't have permission)

# Or, to install it from a local clone:
git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2