Spaces:
Sleeping
Sleeping
Refactor : 에러 나는 부분 예측코드
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ from matplotlib import gridspec
|
|
4 |
import matplotlib.pyplot as plt
|
5 |
import numpy as np
|
6 |
from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
|
|
|
7 |
import torch
|
8 |
import tensorflow as tf
|
9 |
from PIL import ImageDraw
|
|
|
4 |
import matplotlib.pyplot as plt
|
5 |
import numpy as np
|
6 |
from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
|
7 |
+
from transformers import DetrImageProcessor, DetrForObjectDetection
|
8 |
import torch
|
9 |
import tensorflow as tf
|
10 |
from PIL import ImageDraw
|