Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Fix] Fix visualizor #3154

Merged
merged 2 commits into from Jul 3, 2023
Merged

[Fix] Fix visualizor #3154

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
35c2d13
--other=fix show
xiexinch Jun 30, 2023
bba951d
--fix=fix save
xiexinch Jun 30, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
9 changes: 4 additions & 5 deletions demo/inference_demo.ipynb
View file
Open in desktop
Expand Up @@ -21,7 +21,6 @@
"outputs": [],
"source": [
"import torch\n",
"import mmcv\n",
"import matplotlib.pyplot as plt\n",
"from mmengine.model.utils import revert_sync_batchnorm\n",
"from mmseg.apis import init_model, inference_model, show_result_pyplot"
Expand All @@ -48,7 +47,7 @@
"outputs": [],
"source": [
"# build the model from a config file and a checkpoint file\n",
"model = init_model(config_file, checkpoint_file, device='cuda:0')"
"model = init_model(config_file, checkpoint_file, device='cpu')"
]
},
{
Expand All @@ -71,8 +70,8 @@
"outputs": [],
"source": [
"# show the results\n",
"vis_result = show_result_pyplot(model, img, result)\n",
"plt.imshow(mmcv.bgr2rgb(vis_result))"
"vis_result = show_result_pyplot(model, img, result, show=False)\n",
"plt.imshow(vis_result)"
]
},
{
Expand All @@ -99,7 +98,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.9"
"version": "3.10.11"
},
"pycharm": {
"stem_cell": {
Expand Down
2 changes: 1 addition & 1 deletion mmseg/apis/inference.py
View file
Open in desktop
Expand Up @@ -187,7 +187,7 @@ def show_result_pyplot(model: BaseSegmentor,
if hasattr(model, 'module'):
model = model.module
if isinstance(img, str):
image = mmcv.imread(img)
image = mmcv.imread(img, channel_order='rgb')
else:
image = img
if save_dir is not None:
Expand Down
14 changes: 7 additions & 7 deletions mmseg/visualization/local_visualizer.py
View file
Open in desktop
Expand Up @@ -108,14 +108,14 @@ def _draw_sem_seg(self, image: np.ndarray, sem_seg: PixelData,

colors = [palette[label] for label in labels]

self.set_image(image)

# draw semantic masks
mask = np.zeros_like(image, dtype=np.uint8)
for label, color in zip(labels, colors):
self.draw_binary_masks(
sem_seg == label, colors=[color], alphas=self.alpha)
mask[sem_seg[0] == label, :] = color

return self.get_image()
color_seg = (image * (1 - self.alpha) + mask * self.alpha).astype(
np.uint8)
self.set_image(color_seg)
return color_seg

def set_dataset_meta(self,
classes: Optional[List] = None,
Expand Down Expand Up @@ -226,6 +226,6 @@ def add_datasample(
self.show(drawn_img, win_name=name, wait_time=wait_time)

if out_file is not None:
mmcv.imwrite(mmcv.bgr2rgb(drawn_img), out_file)
mmcv.imwrite(mmcv.rgb2bgr(drawn_img), out_file)
else:
self.add_image(name, drawn_img, step)