datachain 0.6.11__py3-none-any.whl → 0.7.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of datachain might be problematic. Click here for more details.
- datachain/__init__.py +1 -2
- datachain/asyn.py +36 -4
- datachain/data_storage/warehouse.py +4 -1
- datachain/lib/dc.py +6 -1
- datachain/lib/file.py +5 -0
- datachain/lib/settings.py +11 -1
- datachain/lib/udf.py +45 -18
- datachain/model/__init__.py +6 -0
- datachain/model/bbox.py +102 -0
- datachain/model/pose.py +88 -0
- datachain/model/segment.py +47 -0
- datachain/model/ultralytics/__init__.py +27 -0
- datachain/model/ultralytics/bbox.py +147 -0
- datachain/model/ultralytics/pose.py +113 -0
- datachain/model/ultralytics/segment.py +91 -0
- datachain/node.py +1 -1
- datachain/query/dataset.py +25 -27
- datachain/toolkit/split.py +6 -2
- {datachain-0.6.11.dist-info → datachain-0.7.1.dist-info}/METADATA +72 -71
- {datachain-0.6.11.dist-info → datachain-0.7.1.dist-info}/RECORD +24 -20
- {datachain-0.6.11.dist-info → datachain-0.7.1.dist-info}/WHEEL +1 -1
- datachain/lib/models/__init__.py +0 -5
- datachain/lib/models/bbox.py +0 -45
- datachain/lib/models/pose.py +0 -37
- datachain/lib/models/yolo.py +0 -39
- {datachain-0.6.11.dist-info → datachain-0.7.1.dist-info}/LICENSE +0 -0
- {datachain-0.6.11.dist-info → datachain-0.7.1.dist-info}/entry_points.txt +0 -0
- {datachain-0.6.11.dist-info → datachain-0.7.1.dist-info}/top_level.txt +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
datachain/__init__.py,sha256=
|
|
1
|
+
datachain/__init__.py,sha256=OGzc8xZWtwqxiiutjU4AxCRPY0lrX_csgERiTrq4G0o,908
|
|
2
2
|
datachain/__main__.py,sha256=hG3Y4ARGEqe1AWwNMd259rBlqtphx1Wk39YbueQ0yV8,91
|
|
3
|
-
datachain/asyn.py,sha256=
|
|
3
|
+
datachain/asyn.py,sha256=5aKrjnUxk0mtnZeFKNJd1DCE0MsnSoyJBZkr0y9H_a0,9313
|
|
4
4
|
datachain/cache.py,sha256=s0YHN7qurmQv-eC265TjeureK84TebWWAnL07cxchZQ,2997
|
|
5
5
|
datachain/cli.py,sha256=hdVt_HJumQVgtaBAtBVJm-uPyYVogMXNVLmRcZyWHgk,36677
|
|
6
6
|
datachain/cli_utils.py,sha256=jrn9ejGXjybeO1ur3fjdSiAyCHZrX0qsLLbJzN9ErPM,2418
|
|
@@ -9,7 +9,7 @@ datachain/dataset.py,sha256=0IN-5y723y-bnFlieKtOFZLCjwX_yplFo3q0DV7LRPw,14821
|
|
|
9
9
|
datachain/error.py,sha256=bxAAL32lSeMgzsQDEHbGTGORj-mPzzpCRvWDPueJNN4,1092
|
|
10
10
|
datachain/job.py,sha256=Jt4sNutMHJReaGsj3r3scueN5aESLGfhimAa8pUP7Is,1271
|
|
11
11
|
datachain/listing.py,sha256=TgKg25ZWAP5enzKgw2_2GUPJVdnQUh6uySHB5SJrUY4,7773
|
|
12
|
-
datachain/node.py,sha256=
|
|
12
|
+
datachain/node.py,sha256=o8Sqy92QkzzcLK6XmIFLyDSE6Rw6kUTmGRhEmfLFdhg,5211
|
|
13
13
|
datachain/nodes_fetcher.py,sha256=F-73-h19HHNGtHFBGKk7p3mc0ALm4a9zGnzhtuUjnp4,1107
|
|
14
14
|
datachain/nodes_thread_pool.py,sha256=uPo-xl8zG5m9YgODjPFBpbcqqHjI-dcxH87yAbj_qco,3192
|
|
15
15
|
datachain/progress.py,sha256=5KotcvvzAUL_RF0GEj4JY0IB1lyImnmHxe89YkT1XO4,4330
|
|
@@ -37,14 +37,14 @@ datachain/data_storage/metastore.py,sha256=5b7o_CSHC2djottebYn-Hq5q0yaSLOKPIRCna
|
|
|
37
37
|
datachain/data_storage/schema.py,sha256=scANMQqozita3HjEtq7eupMgh6yYkrZHoXtfuL2RoQg,9879
|
|
38
38
|
datachain/data_storage/serializer.py,sha256=6G2YtOFqqDzJf1KbvZraKGXl2XHZyVml2krunWUum5o,927
|
|
39
39
|
datachain/data_storage/sqlite.py,sha256=CspRUlYsIcubgzvcQxTACnmcuKESSLZcqCl0dcrtRiA,27471
|
|
40
|
-
datachain/data_storage/warehouse.py,sha256=
|
|
40
|
+
datachain/data_storage/warehouse.py,sha256=yXNU0U3exzR1E6dqbYYmL4RhXWsbYWVdZ3jONGcVniY,30914
|
|
41
41
|
datachain/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
42
42
|
datachain/lib/arrow.py,sha256=-hu9tic79a01SY2UBqkA3U6wUr6tnE3T3q5q_BnO93A,9156
|
|
43
43
|
datachain/lib/clip.py,sha256=lm5CzVi4Cj1jVLEKvERKArb-egb9j1Ls-fwTItT6vlI,6150
|
|
44
44
|
datachain/lib/data_model.py,sha256=dau4AlZBhOFvF7pEKMeqCeRkcFFg5KFvTBWW_2CdH5g,2371
|
|
45
45
|
datachain/lib/dataset_info.py,sha256=q0EW9tj5jXGSD9Lzct9zbH4P1lfIGd_cIWqhnMxv7Q0,2464
|
|
46
|
-
datachain/lib/dc.py,sha256=
|
|
47
|
-
datachain/lib/file.py,sha256
|
|
46
|
+
datachain/lib/dc.py,sha256=u0RQJPG0zwxsoYS-4wrbDBPuLYZajwIi1YX37khKfkI,87942
|
|
47
|
+
datachain/lib/file.py,sha256=-XMkL6ED1sE7TMhWoMRTEuOXswZJw8X6AEmJDONFP74,15019
|
|
48
48
|
datachain/lib/hf.py,sha256=BW2NPpqxkpPwkSaGlppT8Rbs8zPpyYC-tR6htY08c-0,5817
|
|
49
49
|
datachain/lib/image.py,sha256=AMXYwQsmarZjRbPCZY3M1jDsM2WAB_b3cTY4uOIuXNU,2675
|
|
50
50
|
datachain/lib/listing.py,sha256=cVkCp7TRVpcZKSx-Bbk9t51bQI9Mw0o86W6ZPhAsuzM,3667
|
|
@@ -52,11 +52,11 @@ datachain/lib/listing_info.py,sha256=9ua40Hw0aiQByUw3oAEeNzMavJYfW0Uhe8YdCTK-m_g
|
|
|
52
52
|
datachain/lib/meta_formats.py,sha256=anK2bDVbaeCCh0yvKUBaW2MVos3zRgdaSV8uSduzPcU,6680
|
|
53
53
|
datachain/lib/model_store.py,sha256=DNIv8Y6Jtk1_idNLzIpsThOsdW2BMAudyUCbPUcgcxk,2515
|
|
54
54
|
datachain/lib/pytorch.py,sha256=W-ARi2xH1f1DUkVfRuerW-YWYgSaJASmNCxtz2lrJGI,6072
|
|
55
|
-
datachain/lib/settings.py,sha256=
|
|
55
|
+
datachain/lib/settings.py,sha256=ZELRCTLbi5vzRPiDX6cQ9LLg9TefJ_A05gIGni0lll8,2535
|
|
56
56
|
datachain/lib/signal_schema.py,sha256=xwkE5bxJxUhZTjrA6jqN87XbSXPikCbL6eOPL9WyrKM,24556
|
|
57
57
|
datachain/lib/tar.py,sha256=3WIzao6yD5fbLqXLTt9GhPGNonbFIs_fDRu-9vgLgsA,1038
|
|
58
58
|
datachain/lib/text.py,sha256=UNHm8fhidk7wdrWqacEWaA6I9ykfYqarQ2URby7jc7M,1261
|
|
59
|
-
datachain/lib/udf.py,sha256
|
|
59
|
+
datachain/lib/udf.py,sha256=-j0krjNAELTqRI0dB1N65AmawtcIY5vN---AuUcW8Us,13637
|
|
60
60
|
datachain/lib/udf_signature.py,sha256=GXw24A-Olna6DWCdgy2bC-gZh_gLGPQ-KvjuI6pUjC0,7281
|
|
61
61
|
datachain/lib/utils.py,sha256=6NwgWLl5JrgtD4rsSFEe-yR2ntEwJMJEtAZ3FIxK3fg,1529
|
|
62
62
|
datachain/lib/vfile.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
@@ -71,13 +71,17 @@ datachain/lib/convert/values_to_tuples.py,sha256=varRCnSMT_pZmHznrd2Yi05qXLLz_v9
|
|
|
71
71
|
datachain/lib/func/__init__.py,sha256=wlAKhGV0QDg9y7reSwoUF8Vicfqh_YOUNIXLzxICGz4,403
|
|
72
72
|
datachain/lib/func/aggregate.py,sha256=H1ziFQdaK9zvnxvttfnEzkkyGvEEmMAvmgCsBV6nfm8,10917
|
|
73
73
|
datachain/lib/func/func.py,sha256=HAJZ_tpiRG2R-et7pr0WnoyNZYtpbPn3_HBuL3RQpbU,4800
|
|
74
|
-
datachain/
|
|
75
|
-
datachain/
|
|
76
|
-
datachain/
|
|
77
|
-
datachain/
|
|
74
|
+
datachain/model/__init__.py,sha256=R9faX5OHV1xh2EW-g2MPedwbtEqt3LodJRyluB-QylI,189
|
|
75
|
+
datachain/model/bbox.py,sha256=LLtzc8OiL-cxqqlPWXA4MFTo8HRS3GW2gOxA0Sf_cxI,3158
|
|
76
|
+
datachain/model/pose.py,sha256=0URrnS99Ugq0yspCXC2z-hgpybEA5tWLJXpxqVLnAlI,3088
|
|
77
|
+
datachain/model/segment.py,sha256=dMxtm-05fNseEoEKpZj9iDN7fwGK1udyAreN-V-cRks,1597
|
|
78
|
+
datachain/model/ultralytics/__init__.py,sha256=EvcNX9qUyxKXXlKCPpsXeRrabyXk5E9EkN-tyiYkfS4,750
|
|
79
|
+
datachain/model/ultralytics/bbox.py,sha256=OZ9XBdyMOYc401P-RhfSN9QaYvMpnx2Phu9ptaJgZBY,4316
|
|
80
|
+
datachain/model/ultralytics/pose.py,sha256=71KBTcoST2wcEtsyGXqLVpvUtqbp9gwZGA15pEPtX5A,2959
|
|
81
|
+
datachain/model/ultralytics/segment.py,sha256=Z1ab0tZRJubSYNH4KkFlzhYeGNTfAyC71KmkQcToHDQ,2760
|
|
78
82
|
datachain/query/__init__.py,sha256=7DhEIjAA8uZJfejruAVMZVcGFmvUpffuZJwgRqNwe-c,263
|
|
79
83
|
datachain/query/batch.py,sha256=5fEhORFe7li12SdYddaSK3LyqksMfCHhwN1_A6TfsA4,3485
|
|
80
|
-
datachain/query/dataset.py,sha256=
|
|
84
|
+
datachain/query/dataset.py,sha256=sQny-ZemB2HueC4mPg-7qSaqUD85MMO-DQyVVP8K1CA,53765
|
|
81
85
|
datachain/query/dispatch.py,sha256=wjjTWw6sFQbB9SKRh78VbfvwSMgJXCfqJklS3-9KnCU,12025
|
|
82
86
|
datachain/query/metrics.py,sha256=r5b0ygYhokbXp8Mg3kCH8iFSRw0jxzyeBe-C-J_bKFc,938
|
|
83
87
|
datachain/query/params.py,sha256=O_j89mjYRLOwWNhYZl-z7mi-rkdP7WyFmaDufsdTryE,863
|
|
@@ -104,11 +108,11 @@ datachain/sql/sqlite/base.py,sha256=aHSZVvh4XSVkvZ07h3jMoRlHI4sWD8y3SnmGs9xMG9Y,
|
|
|
104
108
|
datachain/sql/sqlite/types.py,sha256=lPXS1XbkmUtlkkiRxy_A_UzsgpPv2VSkXYOD4zIHM4w,1734
|
|
105
109
|
datachain/sql/sqlite/vector.py,sha256=ncW4eu2FlJhrP_CIpsvtkUabZlQdl2D5Lgwy_cbfqR0,469
|
|
106
110
|
datachain/toolkit/__init__.py,sha256=eQ58Q5Yf_Fgv1ZG0IO5dpB4jmP90rk8YxUWmPc1M2Bo,68
|
|
107
|
-
datachain/toolkit/split.py,sha256=
|
|
111
|
+
datachain/toolkit/split.py,sha256=ZgDcrNiKiPXZmKD591_1z9qRIXitu5zwAsoVPB7ykiU,2508
|
|
108
112
|
datachain/torch/__init__.py,sha256=gIS74PoEPy4TB3X6vx9nLO0Y3sLJzsA8ckn8pRWihJM,579
|
|
109
|
-
datachain-0.
|
|
110
|
-
datachain-0.
|
|
111
|
-
datachain-0.
|
|
112
|
-
datachain-0.
|
|
113
|
-
datachain-0.
|
|
114
|
-
datachain-0.
|
|
113
|
+
datachain-0.7.1.dist-info/LICENSE,sha256=8DnqK5yoPI_E50bEg_zsHKZHY2HqPy4rYN338BHQaRA,11344
|
|
114
|
+
datachain-0.7.1.dist-info/METADATA,sha256=9ICI9nDBKNq39JJR2q_RxuYBCFkUD4o81T2FEO8LKDU,18006
|
|
115
|
+
datachain-0.7.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
|
|
116
|
+
datachain-0.7.1.dist-info/entry_points.txt,sha256=0GMJS6B_KWq0m3VT98vQI2YZodAMkn4uReZ_okga9R4,49
|
|
117
|
+
datachain-0.7.1.dist-info/top_level.txt,sha256=lZPpdU_2jJABLNIg2kvEOBi8PtsYikbN1OdMLHk8bTg,10
|
|
118
|
+
datachain-0.7.1.dist-info/RECORD,,
|
datachain/lib/models/__init__.py
DELETED
datachain/lib/models/bbox.py
DELETED
|
@@ -1,45 +0,0 @@
|
|
|
1
|
-
from typing import Optional
|
|
2
|
-
|
|
3
|
-
from pydantic import Field
|
|
4
|
-
|
|
5
|
-
from datachain.lib.data_model import DataModel
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
class BBox(DataModel):
|
|
9
|
-
"""
|
|
10
|
-
A data model for representing bounding boxes.
|
|
11
|
-
|
|
12
|
-
Attributes:
|
|
13
|
-
title (str): The title of the bounding box.
|
|
14
|
-
x1 (float): The x-coordinate of the top-left corner of the bounding box.
|
|
15
|
-
y1 (float): The y-coordinate of the top-left corner of the bounding box.
|
|
16
|
-
x2 (float): The x-coordinate of the bottom-right corner of the bounding box.
|
|
17
|
-
y2 (float): The y-coordinate of the bottom-right corner of the bounding box.
|
|
18
|
-
|
|
19
|
-
The bounding box is defined by two points:
|
|
20
|
-
- (x1, y1): The top-left corner of the box.
|
|
21
|
-
- (x2, y2): The bottom-right corner of the box.
|
|
22
|
-
"""
|
|
23
|
-
|
|
24
|
-
title: str = Field(default="")
|
|
25
|
-
x1: float = Field(default=0)
|
|
26
|
-
y1: float = Field(default=0)
|
|
27
|
-
x2: float = Field(default=0)
|
|
28
|
-
y2: float = Field(default=0)
|
|
29
|
-
|
|
30
|
-
@staticmethod
|
|
31
|
-
def from_xywh(bbox: list[float], title: Optional[str] = None) -> "BBox":
|
|
32
|
-
"""
|
|
33
|
-
Converts a bounding box in (x, y, width, height) format
|
|
34
|
-
to a BBox data model instance.
|
|
35
|
-
|
|
36
|
-
Args:
|
|
37
|
-
bbox (list[float]): A bounding box, represented as a list
|
|
38
|
-
of four floats [x, y, width, height].
|
|
39
|
-
|
|
40
|
-
Returns:
|
|
41
|
-
BBox2D: An instance of the BBox data model.
|
|
42
|
-
"""
|
|
43
|
-
assert len(bbox) == 4, f"Bounding box must have 4 elements, got f{len(bbox)}"
|
|
44
|
-
x, y, w, h = bbox
|
|
45
|
-
return BBox(title=title or "", x1=x, y1=y, x2=x + w, y2=y + h)
|
datachain/lib/models/pose.py
DELETED
|
@@ -1,37 +0,0 @@
|
|
|
1
|
-
from pydantic import Field
|
|
2
|
-
|
|
3
|
-
from datachain.lib.data_model import DataModel
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
class Pose(DataModel):
|
|
7
|
-
"""
|
|
8
|
-
A data model for representing pose keypoints.
|
|
9
|
-
|
|
10
|
-
Attributes:
|
|
11
|
-
x (list[float]): The x-coordinates of the keypoints.
|
|
12
|
-
y (list[float]): The y-coordinates of the keypoints.
|
|
13
|
-
|
|
14
|
-
The keypoints are represented as lists of x and y coordinates, where each index
|
|
15
|
-
corresponds to a specific body part.
|
|
16
|
-
"""
|
|
17
|
-
|
|
18
|
-
x: list[float] = Field(default=None)
|
|
19
|
-
y: list[float] = Field(default=None)
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
class Pose3D(DataModel):
|
|
23
|
-
"""
|
|
24
|
-
A data model for representing 3D pose keypoints.
|
|
25
|
-
|
|
26
|
-
Attributes:
|
|
27
|
-
x (list[float]): The x-coordinates of the keypoints.
|
|
28
|
-
y (list[float]): The y-coordinates of the keypoints.
|
|
29
|
-
visible (list[float]): The visibility of the keypoints.
|
|
30
|
-
|
|
31
|
-
The keypoints are represented as lists of x, y, and visibility values,
|
|
32
|
-
where each index corresponds to a specific body part.
|
|
33
|
-
"""
|
|
34
|
-
|
|
35
|
-
x: list[float] = Field(default=None)
|
|
36
|
-
y: list[float] = Field(default=None)
|
|
37
|
-
visible: list[float] = Field(default=None)
|
datachain/lib/models/yolo.py
DELETED
|
@@ -1,39 +0,0 @@
|
|
|
1
|
-
"""
|
|
2
|
-
This module contains the YOLO models.
|
|
3
|
-
|
|
4
|
-
YOLO stands for "You Only Look Once", a family of object detection models that
|
|
5
|
-
are designed to be fast and accurate. The models are trained to detect objects
|
|
6
|
-
in images by dividing the image into a grid and predicting the bounding boxes
|
|
7
|
-
and class probabilities for each grid cell.
|
|
8
|
-
|
|
9
|
-
More information about YOLO can be found here:
|
|
10
|
-
- https://pjreddie.com/darknet/yolo/
|
|
11
|
-
- https://docs.ultralytics.com/
|
|
12
|
-
"""
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
class PoseBodyPart:
|
|
16
|
-
"""
|
|
17
|
-
An enumeration of body parts for YOLO pose keypoints.
|
|
18
|
-
|
|
19
|
-
More information about the body parts can be found here:
|
|
20
|
-
https://docs.ultralytics.com/tasks/pose/
|
|
21
|
-
"""
|
|
22
|
-
|
|
23
|
-
nose = 0
|
|
24
|
-
left_eye = 1
|
|
25
|
-
right_eye = 2
|
|
26
|
-
left_ear = 3
|
|
27
|
-
right_ear = 4
|
|
28
|
-
left_shoulder = 5
|
|
29
|
-
right_shoulder = 6
|
|
30
|
-
left_elbow = 7
|
|
31
|
-
right_elbow = 8
|
|
32
|
-
left_wrist = 9
|
|
33
|
-
right_wrist = 10
|
|
34
|
-
left_hip = 11
|
|
35
|
-
right_hip = 12
|
|
36
|
-
left_knee = 13
|
|
37
|
-
right_knee = 14
|
|
38
|
-
left_ankle = 15
|
|
39
|
-
right_ankle = 16
|
|
File without changes
|
|
File without changes
|
|
File without changes
|