28 lines
3.3 KiB
Plaintext
28 lines
3.3 KiB
Plaintext
|
o
|
|||
|
|
{<7B>g<EFBFBD>
|
|||
|
|
<00> @ sj d dl Z d dlmZmZ d dlmZ d dlmZ G dd<06> d<06>ZG dd<08> d<08>Z dd
|
|||
|
|
ejd ee fd d
<0A>Z
|
|||
|
|
dS )<0F> N)<02>Image<67> ImageDraw)<01>List)<01>Pathc @ s, e Zd Zdedefdd<04>Zdefdd<07>ZdS ) <09>Point<6E>x<>yc C s t |<01>| _t |<02>| _d S <00>N)<03>intr r )<03>selfr r <00> r <00>>/Users/Leehy/Code/TexTeller/texteller/models/det_model/Bbox.py<70>__init__ s
|
|||
|
|
zPoint.__init__<5F>returnc C s d| j <00> d| j<01> d<03>S )NzPoint(x=z, y=<3D>))r r <00>r r r r
<00>__repr__
s zPoint.__repr__N)<07>__name__<5F>
|
|||
|
|
__module__<EFBFBD> __qualname__r
|
|||
|
|
r <00>strr r r r r
r s r c @ s<> e Zd ZdZddededefdd<08>Zed efd
|
|||
|
|
d <0B><04>Z ed efd d
<0A><04>Z
|
|||
|
|
ed efdd<0F><04>Z ed efdd<11><04>Z d e
fdd<13>Zd e
fdd<15>Zd efdd<17>ZdS )<19>Bboxg<78><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?Nr <00>label<65>
|
|||
|
|
confidence<EFBFBD>contentc C s6 t ||<02>| _t|<03>| _t|<04>| _|| _|| _|| _d S r )r <00>pr
|
|||
|
|
<00>h<>wr r r )r r r r r r r r r r r
r s
|
|||
|
|
|
|||
|
|
|
|||
|
|
z
Bbox.__init__r c C s | j S r )r r r r r
<00>ul_point s z
Bbox.ul_pointc C s t | jj| j | jj<04>S r )r r r r r r r r r
<00>ur_point <00> z
Bbox.ur_pointc C s t | jj| jj| j <00>S r )r r r r r r r r r
<00>ll_point$ r z
Bbox.ll_pointc C s t | jj| j | jj| j <00>S r )r r r r r r r r r r
<00>lr_point( s z
Bbox.lr_pointc C s<> | j j|j jkr| jj|jjks | j j|j jkr"| jj|jjkr"dS | jj|j jks2| j j|jjkr4dS dt| j j|j j <00> t| j|j<05> | jk S )NTFg <00>?)r r r! <00>abs<62>maxr <00>
|
|||
|
|
THREADHOLD<EFBFBD>r <00>otherr r r
<00>same_row- s ,z
Bbox.same_rowc C s* | <00> |<01>s
| jj|jjk S | jj|jjk S )z8
|
|||
|
|
from top to bottom, from left to right
|
|||
|
|
)r( r r r r& r r r
<00>__lt__7 s
|
|||
|
|
z Bbox.__lt__c
C s6 d| j <00> d| j<01> d| j<02> d| j<03> d| j<04> d| j<05> d<07>
S )NzBbox(upper_left_point=z, h=z, w=z ), label=z , confident=z
|
|||
|
|
, content=r )r r r r r r r r r r
r @ s 6z
Bbox.__repr__)Nr N)r r r r% r <00>floatr <00>propertyr r r r! r" <00>boolr( r) r r r r r
r s
|
|||
|
|
r <00>annotated_image.png<6E>img<6D>bboxesc C s<> t t<01><02> <00>}|d }|jdd<03> t<04>| <00>}|D ]F}|jj}|jj}|jj|j } |jj|j
|
|||
|
|
}
|
|||
|
|
|j ||| |
|
|||
|
|
gddd<06> |j rI|j
||f|j dd<08> |jr]|j
||
|
|||
|
|
d f|jd d <09> d
|
|||
|
|
d<08> q| <00>|| <00> d S ) N<>logsT)<01>exist_ok<6F>green<65> )<02>outline<6E>width<74>blue)<01>fill<6C>
|
|||
|
|
<00>red)r <00>os<6F>getcwd<77>mkdirr <00>Drawr r r r r <00> rectangler <00>textr <00>save) r. r/ <00>nameZ
curr_work_dirZlog_dirZdrawer<65>bbox<6F>left<66>top<6F>right<68>bottomr r r
<00> draw_bboxesD s
|
|||
|
|
"<02>rG )r- ) r: <00>PILr r <00>typingr <00>pathlibr r r rG r r r r
<00><module> s 3
|