|
60 | 60 | "\n",
|
61 | 61 | "from colour_checker_detection import (\n",
|
62 | 62 | " ROOT_RESOURCES_EXAMPLES,\n",
|
63 |
| - " detect_colour_checkers_segmentation)\n", |
| 63 | + " detect_colour_checkers_segmentation,\n", |
| 64 | + ")\n", |
64 | 65 | "\n",
|
65 | 66 | "colour.plotting.colour_style()\n",
|
66 | 67 | "\n",
|
|
114 | 115 | ],
|
115 | 116 | "source": [
|
116 | 117 | "COLOUR_CHECKER_IMAGE_PATHS = glob.glob(\n",
|
117 |
| - " os.path.join(ROOT_RESOURCES_EXAMPLES, 'detection', '*19*.png'))\n", |
| 118 | + " os.path.join(ROOT_RESOURCES_EXAMPLES, \"detection\", \"*19*.png\")\n", |
| 119 | + ")\n", |
118 | 120 | "\n",
|
119 | 121 | "COLOUR_CHECKER_IMAGES = [\n",
|
120 | 122 | " colour.cctf_decoding(colour.io.read_image(path))\n",
|
121 | 123 | " for path in COLOUR_CHECKER_IMAGE_PATHS\n",
|
122 | 124 | "]\n",
|
123 | 125 | "\n",
|
124 | 126 | "for image in COLOUR_CHECKER_IMAGES:\n",
|
125 |
| - " colour.plotting.plot_image(colour.cctf_encoding(image));" |
| 127 | + " colour.plotting.plot_image(colour.cctf_encoding(image))" |
126 | 128 | ]
|
127 | 129 | },
|
128 | 130 | {
|
|
166 | 168 | "SWATCHES = []\n",
|
167 | 169 | "for image in COLOUR_CHECKER_IMAGES:\n",
|
168 | 170 | " for colour_checker_data in detect_colour_checkers_segmentation(\n",
|
169 |
| - " image, additional_data=True):\n", |
170 |
| - " \n", |
171 |
| - " swatch_colours, swatch_masks, colour_checker_image = (\n", |
172 |
| - " colour_checker_data.values)\n", |
| 171 | + " image, additional_data=True\n", |
| 172 | + " ):\n", |
| 173 | + " swatch_colours, swatch_masks, colour_checker_image = colour_checker_data.values\n", |
173 | 174 | " SWATCHES.append(swatch_colours)\n",
|
174 |
| - " \n", |
| 175 | + "\n", |
175 | 176 | " # Using the additional data to plot the colour checker and masks.\n",
|
176 | 177 | " masks_i = np.zeros(colour_checker_image.shape)\n",
|
177 |
| - " for i, mask in enumerate(swatch_masks):\n", |
178 |
| - " masks_i[mask[0]:mask[1], mask[2]:mask[3], ...] = 1\n", |
179 |
| - " \n", |
| 178 | + " for mask in swatch_masks:\n", |
| 179 | + " masks_i[mask[0] : mask[1], mask[2] : mask[3], ...] = 1\n", |
| 180 | + "\n", |
180 | 181 | " colour.plotting.plot_image(\n",
|
181 |
| - " colour.cctf_encoding(\n", |
182 |
| - " np.clip(colour_checker_image + masks_i * 0.25, 0, 1)));" |
| 182 | + " colour.cctf_encoding(np.clip(colour_checker_image + masks_i * 0.25, 0, 1))\n", |
| 183 | + " )" |
183 | 184 | ]
|
184 | 185 | },
|
185 | 186 | {
|
|
260 | 261 | }
|
261 | 262 | ],
|
262 | 263 | "source": [
|
263 |
| - "D65 = colour.CCS_ILLUMINANTS['CIE 1931 2 Degree Standard Observer']['D65']\n", |
| 264 | + "D65 = colour.CCS_ILLUMINANTS[\"CIE 1931 2 Degree Standard Observer\"][\"D65\"]\n", |
264 | 265 | "REFERENCE_COLOUR_CHECKER = colour.CCS_COLOURCHECKERS[\n",
|
265 |
| - " 'ColorChecker24 - After November 2014']\n", |
| 266 | + " \"ColorChecker24 - After November 2014\"\n", |
| 267 | + "]\n", |
266 | 268 | "\n",
|
267 | 269 | "colour_checker_rows = REFERENCE_COLOUR_CHECKER.rows\n",
|
268 | 270 | "colour_checker_columns = REFERENCE_COLOUR_CHECKER.columns\n",
|
|
272 | 274 | "# See https://github.com/colour-science/colour-checker-detection/discussions/59\n",
|
273 | 275 | "# for more information.\n",
|
274 | 276 | "REFERENCE_SWATCHES = colour.XYZ_to_RGB(\n",
|
275 |
| - " colour.xyY_to_XYZ(list(REFERENCE_COLOUR_CHECKER.data.values())),\n", |
276 |
| - " 'sRGB', REFERENCE_COLOUR_CHECKER.illuminant)\n", |
| 277 | + " colour.xyY_to_XYZ(list(REFERENCE_COLOUR_CHECKER.data.values())),\n", |
| 278 | + " \"sRGB\",\n", |
| 279 | + " REFERENCE_COLOUR_CHECKER.illuminant,\n", |
| 280 | + ")\n", |
277 | 281 | "\n",
|
278 | 282 | "for i, swatches in enumerate(SWATCHES):\n",
|
279 |
| - " swatches_xyY = colour.XYZ_to_xyY(colour.RGB_to_XYZ(\n", |
280 |
| - " swatches, 'sRGB', D65))\n", |
| 283 | + " swatches_xyY = colour.XYZ_to_xyY(colour.RGB_to_XYZ(swatches, \"sRGB\", D65))\n", |
281 | 284 | "\n",
|
282 | 285 | " colour_checker = colour.characterisation.ColourChecker(\n",
|
283 | 286 | " os.path.basename(COLOUR_CHECKER_IMAGE_PATHS[i]),\n",
|
284 | 287 | " dict(zip(REFERENCE_COLOUR_CHECKER.data.keys(), swatches_xyY)),\n",
|
285 |
| - " D65, colour_checker_rows, colour_checker_columns)\n", |
286 |
| - " \n", |
| 288 | + " D65,\n", |
| 289 | + " colour_checker_rows,\n", |
| 290 | + " colour_checker_columns,\n", |
| 291 | + " )\n", |
| 292 | + "\n", |
287 | 293 | " colour.plotting.plot_multi_colour_checkers(\n",
|
288 |
| - " [REFERENCE_COLOUR_CHECKER, colour_checker])\n", |
289 |
| - " \n", |
| 294 | + " [REFERENCE_COLOUR_CHECKER, colour_checker]\n", |
| 295 | + " )\n", |
| 296 | + "\n", |
290 | 297 | " swatches_f = colour.colour_correction(swatches, swatches, REFERENCE_SWATCHES)\n",
|
291 |
| - " swatches_f_xyY = colour.XYZ_to_xyY(colour.RGB_to_XYZ(\n", |
292 |
| - " swatches_f, 'sRGB', D65))\n", |
| 298 | + " swatches_f_xyY = colour.XYZ_to_xyY(colour.RGB_to_XYZ(swatches_f, \"sRGB\", D65))\n", |
293 | 299 | " colour_checker = colour.characterisation.ColourChecker(\n",
|
294 |
| - " '{0} - CC'.format(os.path.basename(COLOUR_CHECKER_IMAGE_PATHS[i])),\n", |
| 300 | + " f\"{os.path.basename(COLOUR_CHECKER_IMAGE_PATHS[i])} - CC\",\n", |
295 | 301 | " dict(zip(REFERENCE_COLOUR_CHECKER.data.keys(), swatches_f_xyY)),\n",
|
296 |
| - " D65, colour_checker_rows, colour_checker_columns)\n", |
297 |
| - " \n", |
| 302 | + " D65,\n", |
| 303 | + " colour_checker_rows,\n", |
| 304 | + " colour_checker_columns,\n", |
| 305 | + " )\n", |
| 306 | + "\n", |
298 | 307 | " colour.plotting.plot_multi_colour_checkers(\n",
|
299 |
| - " [REFERENCE_COLOUR_CHECKER, colour_checker])\n", |
| 308 | + " [REFERENCE_COLOUR_CHECKER, colour_checker]\n", |
| 309 | + " )\n", |
300 | 310 | "\n",
|
301 |
| - " colour.plotting.plot_image(colour.cctf_encoding(\n", |
302 |
| - " colour.colour_correction(\n", |
303 |
| - " COLOUR_CHECKER_IMAGES[i], swatches, REFERENCE_SWATCHES)));" |
| 311 | + " colour.plotting.plot_image(\n", |
| 312 | + " colour.cctf_encoding(\n", |
| 313 | + " colour.colour_correction(\n", |
| 314 | + " COLOUR_CHECKER_IMAGES[i], swatches, REFERENCE_SWATCHES\n", |
| 315 | + " )\n", |
| 316 | + " )\n", |
| 317 | + " )" |
304 | 318 | ]
|
305 | 319 | },
|
306 | 320 | {
|
|
414 | 428 | ],
|
415 | 429 | "source": [
|
416 | 430 | "for image in COLOUR_CHECKER_IMAGES:\n",
|
417 |
| - " for colour_checker_data in detect_colour_checkers_segmentation(\n", |
418 |
| - " image, show=True):\n", |
419 |
| - " pass" |
| 431 | + " for _colour_checker_data in detect_colour_checkers_segmentation(image, show=True):\n", |
| 432 | + " ..." |
420 | 433 | ]
|
421 | 434 | },
|
422 | 435 | {
|
|
537 | 550 | ")\n",
|
538 | 551 | "\n",
|
539 | 552 | "for path in COLOUR_CHECKER_IMAGE_PATHS:\n",
|
540 |
| - " for colour_checker_data in detect_colour_checkers_segmentation(\n", |
| 553 | + " for _colour_checker_data in detect_colour_checkers_segmentation(\n", |
541 | 554 | " path, apply_cctf_decoding=True, show=True\n",
|
542 | 555 | " ):\n",
|
543 |
| - " pass" |
| 556 | + " ..." |
544 | 557 | ]
|
545 | 558 | }
|
546 | 559 | ],
|
|
0 commit comments