Calling Yolact from another Python file

Hi @dbolya, Thanks again for such a useful model. I am trying to use yolact inside another file and I will use its outputs in terms of masks, bounding-boxes, confidence rates and classes. But it gives errors when I import yolact.eval --> ImportError: cannot import name 'Yolact' If I import yolact, there is no error. If I from yolact import eval --> ImportError: cannot import name 'Yolact', it gives the same error. After import yolact without no errors, I type yolact and it gives <module 'yolact' (namespace)>. But if I import another lib, it shows the path to that. How can I call eval.py from another python file and use its outputs? By the way my config is python 3.6, pytorch 1.0.1, cuda 9.0…

[dbolya]

What does your final structure look like? It sounds like you’re importing yolact.py not importing from the yolact directory. Though that’s actually what you want (importing yolact.py, that is).

Here’s the rough procedure:

import torch
import cv2 # Optional, see below

from yolact import Yolact
from data import set_cfg
from utils.augmentations import FastBaseTransform
from layers.output_utils import postprocess

##### Setup #####

# CUDA setup, I should really just replace all Tensor initializations but I'm in too deep at this point
torch.backends.cudnn.fastest = True
torch.set_default_tensor_type('torch.cuda.FloatTensor')

# Use whichever config you want here.
set_cfg('yolact_base_config')

net = Yolact().cuda()
net.load_weights('path/to/weight/file')
net.eval()

transform = FastBaseTransform()

##### Now to use ######

# Get your image from somewhere (BGR format, [0, 255], numpy)
img = cv2.imread('path/to/image')
h, w, _ = img.shape

batch = transform(torch.from_numpy(img).cuda().float()[None, ...])
preds = net(batch)

# You can modify the score threshold to your liking
classes, scores, boxes, masks = postprocess(preds , w, h, score_threshold=0.15)

Sorry for the number of lines of code necessary to do this. I should probably make an example script and put it on the readme or something. (Also this code is untested, if there are any bugs lmk).

[kre-ka]

@zyxdb

Hey, I have a similar problem: AttributeError: ‘Config’ object has no attribute ‘mask_proto_debug’

I want to know what you said “necessary paths” and “current directory”,I only see two places in the code where I need to fill the path,(weights and img)

For a lack of better solution (or rather inability to find one), I just commented calling mask_proto_debug out in output_utils.py line 63:

#if cfg.mask_proto_debug:
        #    np.save('scripts/proto.npy', proto_data.cpu().numpy())

“Fixed” the problem without any devastating consequences (for now)