Fix export of video task for coco, yolo, PascalVOC #137
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
fuji44
commented
Mar 6, 2023
|
|
||
| def export_coco( | ||
| self, | ||
| project: str, |
Contributor
Author
There was a problem hiding this comment.
タスクを動画と画像のいずれで処理すべきかを判断するために、プロジェクトタイプを用いることにしました。
そのため、SDK利用者に新たにプロジェクトスラグを指定してもらうようにしました。
Contributor
There was a problem hiding this comment.
こちら今後project_slugが必須になると思うので、READMEの方もそれに合わせて変えておいてもらえると助かります！
| } | ||
|
|
||
|
|
||
| def _generate_coco_images( |
Contributor
Author
Comment on lines
+216
to
218
| if annotation_type != AnnotationType.pose_estimation.value and ( | ||
| not points or (len(points) == 0) | ||
| ): |
Comment on lines
+234
to
+240
| def __get_coco_category_by_name(categories: list, name: str) -> Optional[dict]: | ||
| matched_categories = [ | ||
| category for category in categories if category["name"] == name | ||
| ] | ||
| if len(matched_categories) >= 1: | ||
| return matched_categories[0] | ||
| return None |
Comment on lines
+1035
to
+1041
| @contextmanager | ||
| def VideoCapture(*args, **kwds): | ||
| videoCapture = cv2.VideoCapture(*args, **kwds) | ||
| try: | ||
| yield videoCapture | ||
| finally: | ||
| videoCapture.release() |
Contributor
Author
There was a problem hiding this comment.
withを使った自動リソース開放のためContextManagerを使います。
| videoCapture.release() | ||
|
|
||
|
|
||
| def _download_file(url: str, output_file_path: str, chunk_size: int = 8192) -> str: |
Contributor
Author
There was a problem hiding this comment.
チャンクサイズは適切なものが推測しづらいので、ひとまずの値です。将来的には、利用者が指定できるようにしてもよいかもしれません。
Comment on lines
+1086
to
+1089
| with NamedTemporaryFile(prefix="fastlabel-sdk-") as video_file: | ||
| video_file_path = _download_file( | ||
| url=video_task["url"], output_file_path=video_file.name | ||
| ) |
Contributor
Author
There was a problem hiding this comment.
動画ファイルは一時ファイルとして書き込み、withを抜けたら削除されるようにしています。
OSやPythonランタイムによりますが、xnuxでCPythonならおおよそ /tmp/fastlabel-sdk-xxxx になるはずです。
Comment on lines
+63
to
+64
| def is_video_project_type(project_type: str): | ||
| return type(project_type) is str and project_type.startswith("video_") |
Contributor
Author
There was a problem hiding this comment.
文字の前方一致で判断します。
他の類似コードではすべてのタイプを定数化して一致判定していることがほとんどですが、定数の管理コストを考えて前方一致で判定することにしました。問題があるようなら指摘してください。
ueno-k-work
approved these changes
Mar 20, 2023
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
see: https://github.com/fastlabel/fastlabel-application/issues/3475
補足
WebのExportの動作と極力合わせていますが、一部異なります。
問題がありそうなら指摘をお願いします。
特に1。Webではアノテーションがないタスクでもアノテーションファイルを出力していますが、SDKではアノテーションがないタスクのアノテーションファイルは出力しません。また、いくつか既存のバグを発見し影響があるものは修正しましたが、以下は大きな影響がないため ueno さんと相談してコメントを残すのみとしています。
categoriesのcolorがすべて同じになります。テスト
WebとSDKのぞれぞれで、動画タスクをCOCO、YOLO、PascalVOCでエクスポートし、Diffツールで差分を確認しました。