Skip to content

Can't import PostProcess and custom_logger from YOLO #139

Open
@Jaykingamez

Description

@Jaykingamez

Describe the bug

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
[<ipython-input-9-b85eab8d12e7>](https://localhost:8080/#) in <cell line: 12>()
     10 sys.path.append(str(project_root))
     11 
---> 12 from yolo import (
     13     AugmentationComposer,
     14     Config,

ImportError: cannot import name 'PostProccess' from 'yolo' (/content/YOLO/yolo/__init__.py)

To Reproduce

I was trying to follow the YOLO/examples/notebook_inference.ipynb.
Steps to reproduce the behavior: I'm using Google Colab

  1. !pip install git+https://github.com/WongKinYiu/YOLO.git2. Click on '....'
  2. Use the following code
import sys
from pathlib import Path

import torch
from hydra import compose, initialize
from PIL import Image

project_root = Path().resolve().parent
print(project_root)
sys.path.append(str(project_root))

from yolo import (
    AugmentationComposer,
    Config,
    PostProccess,
    create_converter,
    create_model,
    custom_logger,
    draw_bboxes,
)

Expected behavior

I expect PostProcessing to be imported, commenting it away, I also can't import custom_logger.

Screenshots

If applicable, add screenshots to help explain your problem.

System Info (please complete the following ## information):

runtime

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions